Update app.py

app.py

@@ -107,15 +107,18 @@ def generate(
     print("Tokens = "+str([response.token.text for response in stream]))
     output = ""
     previous_token = ""
-    
+    t = 0
     for response in stream:
         if (
             (response.token.text in ["Human", "-----", "Question:"] and previous_token in ["\n", "-----"])
             or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
         ):
+            print("OUT")
             return output.strip()
         else:
             output += response.token.text
+            print(f"Out_{t} : {output}")
+            t += 1
         previous_token = response.token.text
     print("Output = "+str(output.strip()))
     return output.strip()