Update app.py

app.py

@@ -105,11 +105,12 @@ def generate(
     else :
         ValueError("Unsupported version of the Coding assistant")
       
-    print("Tokens = "+str([response.token.text for response in stream]))
+    #print("Tokens = "+str([response.token.text for response in stream]))
     output = ""
     previous_token = ""
     t = 0
     for response in stream:
+        print(f"IN_{t}")
         if (
             (response.token.text in ["Human", "-----", "Question:"] and previous_token in ["\n", "-----"])
             or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
@@ -121,7 +122,7 @@ def generate(
             print(f"Out_{t} : {output}")
             t += 1
         previous_token = response.token.text
-    print("Output = "+str(output.strip()))
+    print("Output = "+str(output))
     return output.strip()