Update app.py

app.py

@@ -7,7 +7,7 @@ from huggingface_hub import login
 
 # Load the model
 model = AutoModel.from_pretrained("jcsagar/CXR-LLAVA-v2", trust_remote_code=True)
-model = model.to("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to("cuda")
 
 # Define the function to generate the report
 def generate_report(image):