Update app.py

app.py

@@ -56,10 +56,10 @@ def _launch_demo(args, model, tokenizer, config):
         
         # Tokenize the input
         input_ids = tokenizer.encode(_query, return_tensors='pt')
-        
+        print("Input IDs:", input_ids)
         # Generate a response using the model
         generated_ids = model.generate(input_ids, max_length=300)
-        
+        print("Generated IDs:", generated_ids)
         # Decode the generated IDs to text
         full_response = tokenizer.decode(generated_ids[0], skip_special_tokens=True)