Update app.py

app.py

@@ -25,12 +25,14 @@ def generate_response(model, tokenizer, prompt):
         pad_token_id=pad_token_id
     )
     response = tokenizer.decode(output[0], skip_special_tokens=True)
+    print("Generated response::", response )
+    print("len(prompt)::", len(prompt) )
     response = response[len(prompt):]
     return response
 
 def predict(input):
     prediction = generate_response(my_model, my_tokenizer, input)
-    print("type of response:", type(input))
+    print("type of response:", type(prediction))
     return prediction