McCoasta commited on
Commit
c1a9fda
·
1 Parent(s): 048c822

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,10 +2,10 @@ from PIL import Image
2
  import requests
3
  import gradio as gr
4
 
5
- gr.Interface.load("models/Salesforce/blip-image-captioning-base").launch()
6
  from transformers import BlipProcessor, BlipForConditionalGeneration
7
 
8
- model_id = Salesforceblip-image-captioning-base
9
  model = BlipForConditionalGeneration.from_pretrained(model_id)
10
  processor = BlipProcessor.from_pretrained(model_id)
11
 
 
2
  import requests
3
  import gradio as gr
4
 
5
+ gr.Interface.load("models/Salesforce/blip-image-captioning-large").launch()
6
  from transformers import BlipProcessor, BlipForConditionalGeneration
7
 
8
+ model_id = Salesforceblip-image-captioning-large
9
  model = BlipForConditionalGeneration.from_pretrained(model_id)
10
  processor = BlipProcessor.from_pretrained(model_id)
11