vicgalle commited on
Commit
095153b
·
1 Parent(s): 1ed0ea9
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ st.title("Image to Text Converter")
8
  uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
9
 
10
 
11
- model = PaliGemmaForConditionalGeneration.from_pretrained("Fer14/paligemma_coffe_describer")
12
  processor = PaliGemmaProcessor.from_pretrained("Fer14/paligemma_coffe_describer")
13
 
14
  prompt = (
 
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 = (