Update app.py

app.py

@@ -15,8 +15,8 @@ def generate_ad_post(image):
     if not isinstance(image, Image.Image):
         image = Image.fromarray(image)
     
-    # Измененный текст запроса для генерации рекламного поста
-    inputs = florence_processor(text="Создайте рекламный пост на русском языке", images=image, return_tensors="pt").to(device)
+    # Более конкретный текст запроса для генерации рекламного поста
+    inputs = florence_processor(text="Напишите рекламный пост на русском языке о продукте на изображении", images=image, return_tensors="pt").to(device)
     generated_ids = florence_model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],
@@ -31,8 +31,11 @@ def generate_ad_post(image):
         task="Создание рекламного поста",
         image_size=(image.width, image.height)
     )
-    prompt =  parsed_answer["Создание рекламного поста"]
-    print("\n\nГенерация завершена!:"+ prompt)
+    
+    # Извлечение текста из ответа
+    prompt = parsed_answer.get("Создание рекламного поста", generated_text)
+    
+    print("\n\nГенерация завершена!:" + prompt)
     return prompt
 
 io = gr.Interface(generate_ad_post,