Spaces:
Sleeping
Sleeping
update modle_id
Browse files
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(
|
| 12 |
-
processor = PaliGemmaProcessor.from_pretrained(
|
| 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"
|