Update app.py

app.py

@@ -87,12 +87,12 @@ def predict(text,
                 x = x.strip()   
                 a, b=   [[y[0],convert_to_markdown(y[1])] for y in history]+[[text, convert_to_markdown(x)]],history + [[text,x]]
                 print("Erzeuge")
-                return
+                return a,b
             if shared_state.interrupted:
                 shared_state.recover()
                 try:
                     print("Erfolg")
-                    return
+                    return a,b
                 except:
                     pass
     del input_ids
@@ -101,6 +101,7 @@ def predict(text,
     
     try:
         print("erfolg")
+        return a,b
     except:
         pass
 
@@ -226,7 +227,9 @@ print("done")
 ##############################################
 #Testen des fine-tuned Modells
 print("Predict")
-predict("Was ist Tis?")
+a,b = predict("Was ist Tis?")
+print(a)
+print(b)
 print("done")
 
 #######################################################################