Update app.py

app.py

@@ -55,16 +55,16 @@ def get_response(input_text):
     #num_beam_groups=num_beams, diversity_penalty=0.5
     tgt_text = tokenizer.batch_decode(translated, skip_special_tokens=True)
     try:
-        adequacy_filtered_phrases = adequacy_score.filter(input_text,tgt_text, adequacy_threshold, device)
-        if len(adequacy_filtered_phrases) > 0 :
-            fluency_filtered_phrases = fluency_score.filter(adequacy_filtered_phrases, fluency_threshold, device )
-            if len(fluency_filtered_phrases) > 0 :
-                diversity_scored_phrases = diversity_score.rank(input_text, fluency_filtered_phrases, diversity_ranker)
-                return get_max_str(diversity_scored_phrases)
+        adequacy_filtered = adequacy_score.filter(input_text,tgt_text, adequacy_threshold, device)
+        if len(adequacy_filtered) > 0 :
+            fluency_filtered_ = fluency_score.filter(adequacy_filtered, fluency_threshold, device )
+            if len(fluency_filtered) > 0 :
+                diversity_scored_phrases = diversity_score.rank(input_text, fluency_filtered, diversity_ranker)
+                return get_max_str(diversity_scored)
             else:
-                return get_max_str(fluency_filtered_phrases)
+                return get_max_str(fluency_filtered)
         else:
-            return get_max_str(adequacy_filtered_phrases)
+            return get_max_str(adequacy_filtered)
     except:
         return(get_max_str(tgt_text))
         
@@ -72,8 +72,8 @@ def get_response(input_text):
 
 import gradio as gr
 
-def get_fun(txt):
-  tokens = splitter.split(text=txt)
+def get_fun(Input_txt):
+  tokens = splitter.split(text=Input_txt)
 
   txt_paraphrase=''
   for phrase in tokens:
@@ -81,7 +81,7 @@ def get_fun(txt):
     txt_paraphrase=txt_paraphrase+' '+tmp
   return txt_paraphrase
 
-iface = gr.Interface(fn=get_fun, inputs="text", outputs="text", title = " Ai Re-Phraser Q'Hackday")
+iface = gr.Interface(fn=get_fun, inputs="text", outputs="text" (lines=5), title = " Ai Re-Phraser Q'Hackday")
 iface.launch(inline=False)
 
 """# New Section"""