Spaces:
Runtime error
Runtime error
ShreyMehra
commited on
Commit
•
5d02c70
1
Parent(s):
e22d5df
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ image = st.file_uploader(label = "Upload your image here",type=['png','jpg','jpe
|
|
17 |
@st.cache
|
18 |
def load_model():
|
19 |
processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
|
20 |
-
model = Blip2ForConditionalGeneration.from_pretrained("Shrey23/Image-Captioning", device_map="auto",
|
21 |
return processor, model
|
22 |
|
23 |
processor, model = load_model() #load model
|
|
|
17 |
@st.cache
|
18 |
def load_model():
|
19 |
processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
|
20 |
+
model = Blip2ForConditionalGeneration.from_pretrained("Shrey23/Image-Captioning", device_map="auto", )
|
21 |
return processor, model
|
22 |
|
23 |
processor, model = load_model() #load model
|