Hugging Face's logo

Spaces:
satyaaa-m
/
image_captioning
Sleeping

App Files Community
satyaaa-m commited on
Commit
807b2d7
·
verified ·
1 Parent(s): bdeb045

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -1,13 +1,11 @@
1
  import gradio as gr
2
  import os
3
 
4
- from transformers import BlipForConditionalGeneration
5
- from transformers import AutoProcessor
6
 
7
- model = BlipForConditionalGeneration.from_pretrained(
8
- "./models/Salesforce/blip-image-captioning-base")
9
- processor = AutoProcessor.from_pretrained(
10
- "./models/Salesforce/blip-image-captioning-base")
11
 
12
 
13
  def caption(image):
 
1
  import gradio as gr
2
  import os
3
 
4
+ # Load model directly
5
+ from transformers import AutoProcessor, AutoModelForImageTextToText
6
 
7
+ processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
8
+ model = AutoModelForImageTextToText.from_pretrained("Salesforce/blip-image-captioning-base")
 
 
9
 
10
 
11
  def caption(image):