koey811 commited on
Commit
f976398
·
verified ·
1 Parent(s): 1f0f5c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ except ImportError:
15
  import torch
16
 
17
  # Load the image captioning model
18
- caption_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
19
 
20
  # Load the text generation model
21
  text_generation_model = AutoModelForCausalLM.from_pretrained("gpt2")
 
15
  import torch
16
 
17
  # Load the image captioning model
18
+ caption_model = pipeline("image-to-text", model="IAMJB/chexpert-mimic-cxr-findings-baseline"")
19
 
20
  # Load the text generation model
21
  text_generation_model = AutoModelForCausalLM.from_pretrained("gpt2")