Update app.py

app.py

@@ -4,12 +4,14 @@ from PIL import Image
 import torch
 import os
 from huggingface_hub import login
+import spaces
 
 # Load the model
 model = AutoModel.from_pretrained("jcsagar/CXR-LLAVA-v2", trust_remote_code=True)
 model = model.to("cuda")
 
 # Define the function to generate the report
+@spaces.GPU
 def generate_report(image):
     image = Image.open(image).convert("RGB")
     response = model.write_radiologic_report(image)