ShreyMehra commited on
Commit
5d02c70
1 Parent(s): e22d5df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", load_in_8bit=True)
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