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"
|