Update app.py

app.py

@@ -82,15 +82,22 @@ def ACE(file):
             output_all.append(output)
             probability_all.append(probability)
 
-    summary = OrderedDict()
-    summary['Seq'] = seq_all
-    summary['Class'] = output_all
-    summary['Probability'] = probability_all
-    summary_df = pd.DataFrame(summary)
-    summary_df.to_csv('output.csv', index=False)
-    return 'output.csv'
+    if seq_len > 30:
+        out_text = "None"
+        out_prob = "None"
+        summary = OrderedDict()
+        summary['Seq'] = seq_all
+        summary['Class'] = output_all
+        summary['Probability'] = probability_all
+        summary_df = pd.DataFrame(summary)
+        summary_df.to_csv('output.csv', index=False)
+    else:
+        out_text = output
+        out_prob = probability
+
+    return 'outputs.csv', out_text, out_prob
 
 iface = gr.Interface(fn=ACE,
                      inputs="text", 
-                     outputs= "file")
+                     outputs= ["file","text","text"])
 iface.launch()