Update app.py

app.py

@@ -246,15 +246,15 @@ for model in models:
     models2.append(loaded_model)
 
 
-def text_it(inputs,outputs):
-    grad.Interface(fn=generate,
-               inputs=inputs,
-               outputs=outputs,
-               examples=examples,
-               allow_flagging='never',
-               cache_examples=False,
-               theme="default").launch(enable_queue=True, debug=True)
-    # return text_gen(inputs)
+def text_it(inputs,text_gen=generate):
+    # grad.Interface(fn=generate,
+    #            inputs=inputs,
+    #            outputs=outputs,
+    #            examples=examples,
+    #            allow_flagging='never',
+    #            cache_examples=False,
+    #            theme="default").launch(enable_queue=True, debug=True)
+    return text_gen(inputs)
 
 
 def set_model(current_model_index):