Spaces:

Fer14
/

coffee_machine_captioning

Sleeping

vicgalle commited on Jun 10, 2024

Commit

095153b

1 Parent(s): 1ed0ea9

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ st.title("Image to Text Converter")
 uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-model = PaliGemmaForConditionalGeneration.from_pretrained("Fer14/paligemma_coffe_describer")
 processor = PaliGemmaProcessor.from_pretrained("Fer14/paligemma_coffe_describer")
 prompt  = (

 uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+model = PaliGemmaForConditionalGeneration.from_pretrained("Fer14/paligemma_coffe_describer", from_tf=True)
 processor = PaliGemmaProcessor.from_pretrained("Fer14/paligemma_coffe_describer")
 prompt  = (