Update app.py

app.py

@@ -47,6 +47,7 @@ Diagnosis:"""
     model.eval()
     with torch.no_grad():
         outputs = model.generate(**inputs, max_new_tokens=4)
+        print("Raw output:", outputs)  # Add this line
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
     diagnosis = decoded.split("Diagnosis:")[-1].strip()
     return diagnosis