Update app.py

app.py

@@ -2,7 +2,7 @@ from PIL import Image
 import gradio as gr
 from transformers import BlipProcessor, BlipForConditionalGeneration
 
-model_id = "Salesforce/blip-image-captioning-large"
+model_id = "Salesforce/blip-image-captioning-base"
 
 model = BlipForConditionalGeneration.from_pretrained(model_id)
 processor = BlipProcessor.from_pretrained(model_id)
@@ -17,8 +17,8 @@ def launch(input_file):
     generated_text = processor.decode(out[0], skip_special_tokens=True)
     
     # Agregar prefijo y sufijo al texto generado
-    prefijo = "Texture of a "
-    sufijo = " top view, flat"
+    prefijo = "Texture, "
+    sufijo = " top view, flat, hq, detailed"
     result = f"{prefijo}{generated_text}{sufijo}"
     
     # Devolver tanto la imagen como el texto