Update app.py

app.py

@@ -52,14 +52,14 @@ def givetext(input_text,lmodel,ltokenizer):
         
         model_input = ltokenizer(eval_prompt, return_tensors="pt").to("cpu")
 
-        PRINT(" BEFORE EVAL LMODEL")
+        print(" BEFORE EVAL LMODEL")
         
         lmodel.eval()
 
-        PRINT("BEFORE DOING TORCH.NO_GRAD()")
+        print("BEFORE DOING TORCH.NO_GRAD()")
         
         with torch.no_grad():
-            PRINT("BEFORE RETURNING")
+            print("BEFORE RETURNING")
             
             return (ltokenizer.decode(lmodel.generate(**model_input, max_new_tokens=1000)[0], skip_special_tokens=True))
             #return (ltokenizer.decode(lmodel.generate(**model_input, max_new_tokens=100)[0], skip_special_tokens=True))
@@ -68,7 +68,7 @@ def givetext(input_text,lmodel,ltokenizer):
         #txt1 = "My name is {fname}, I'm {age}".format(fname = "John", age = 36)
 
 def mental_chat(message, history):
-    PRINT("BEFORE CALLING GIVETEXT")
+    print("BEFORE CALLING GIVETEXT")
     return givetext(message,newmodel,newtokenizer)
 
 demo = gr.ChatInterface(mental_chat)