coeuslearning commited on
Commit
71a7f7f
1 Parent(s): 4004279

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -73,12 +73,12 @@ def generate(
73
  # outputs.append(text)
74
  # concatenated_outputs = " ".join(outputs)
75
  # yield concatenated_outputs
76
- concatenated_outputs = " ".join([text for text in streamer])
77
  # yield concatenated_outputs
78
 
79
  # Mask the output here
80
  masked_output = mask_with_protecto(concatenated_outputs)
81
- yield concatenated_outputs + "\n\n\n\n" + masked_output
82
 
83
 
84
  def mask_with_protecto(text_for_prompt):
@@ -99,8 +99,7 @@ def mask_with_protecto(text_for_prompt):
99
  # Parse the masked result from the API response and format it for display
100
  masked_result = response.json()
101
  masked_result_token_value = str(masked_result["data"][0]["token_value"])
102
- final_result = json.dumps(masked_result, indent=4)
103
- return_value = str(masked_result_token_value)
104
  return(return_value)
105
  else:
106
  # Return an error message if the API request was not successful.
 
73
  # outputs.append(text)
74
  # concatenated_outputs = " ".join(outputs)
75
  # yield concatenated_outputs
76
+ concatenated_outputs = "".join([text for text in streamer])
77
  # yield concatenated_outputs
78
 
79
  # Mask the output here
80
  masked_output = mask_with_protecto(concatenated_outputs)
81
+ yield masked_output
82
 
83
 
84
  def mask_with_protecto(text_for_prompt):
 
99
  # Parse the masked result from the API response and format it for display
100
  masked_result = response.json()
101
  masked_result_token_value = str(masked_result["data"][0]["token_value"])
102
+ return_value = masked_result_token_value
 
103
  return(return_value)
104
  else:
105
  # Return an error message if the API request was not successful.