mariamSoub commited on
Commit
41223ef
·
verified ·
1 Parent(s): 3b7fe2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -113,9 +113,9 @@ def show_neutralized_prompt(input_text):
113
  else:
114
  second = -1
115
  if second != -1:
116
- print(generated_text[0:second])
117
  else:
118
- print(generated_text[0:first])
119
 
120
  # FAIRNESS MODEL (MNLI)
121
  mnli_model_name = "facebookAI/roberta-large-mnli"
 
113
  else:
114
  second = -1
115
  if second != -1:
116
+ return generated_text[first + len(sep):second].strip()
117
  else:
118
+ return generated_text[first + len(sep):].strip()
119
 
120
  # FAIRNESS MODEL (MNLI)
121
  mnli_model_name = "facebookAI/roberta-large-mnli"