jcsagar commited on
Commit
6252940
·
verified ·
1 Parent(s): 960a738

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  import os
6
 
7
  # Load the model
8
- model = AutoModel.from_pretrained("ECOFRI/CXR-LLAVA-v2", trust_remote_code=True)
9
  model = model.to("cuda" if torch.cuda.is_available() else "cpu")
10
 
11
  # Define the function to generate the report
 
5
  import os
6
 
7
  # Load the model
8
+ model = AutoModel.from_pretrained("jcsagar/CXR-LLAVA-v2", trust_remote_code=True)
9
  model = model.to("cuda" if torch.cuda.is_available() else "cpu")
10
 
11
  # Define the function to generate the report