Update app.py

app.py

@@ -3,11 +3,14 @@ from transformers import AutoModel
 from PIL import Image
 import torch
 import os
+from huggingface_hub import login
 
 MODEL_ACCESS_TOKEN = os.getenv('MODEL_ACCESS_TOKEN')
 
+login(MODEL_ACCESS_TOKEN)
+
 # Load the model
-model = AutoModel.from_pretrained("jcsagar/CXR-LLAVA-v2", trust_remote_code=True, token=MODEL_ACCESS_TOKEN)
+model = AutoModel.from_pretrained("jcsagar/CXR-LLAVA-v2", trust_remote_code=True)
 model = model.to("cuda" if torch.cuda.is_available() else "cpu")
 
 # Define the function to generate the report