vicgalle commited on
Commit
a6920b2
1 Parent(s): 095153b

update modle_id

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,9 +7,10 @@ st.title("Image to Text Converter")
7
 
8
  uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
9
 
 
10
 
11
- model = PaliGemmaForConditionalGeneration.from_pretrained("Fer14/paligemma_coffe_describer", from_tf=True)
12
- processor = PaliGemmaProcessor.from_pretrained("Fer14/paligemma_coffe_describer")
13
 
14
  prompt = (
15
  f"Generate a caption for the following coffee maker image. The caption has to be of the following structure:\n"
 
7
 
8
  uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
9
 
10
+ model_id = "Fer14/paligemma_coffe_machine_caption"
11
 
12
+ model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
13
+ processor = PaliGemmaProcessor.from_pretrained(model_id)
14
 
15
  prompt = (
16
  f"Generate a caption for the following coffee maker image. The caption has to be of the following structure:\n"