Update app.py

app.py

@@ -1,19 +1,11 @@
-from PIL import Image
-import requests
 import gradio as gr
-
-from transformers import BlipProcessor, BlipForConditionalGeneration
+from transformers import pipeline
 
 model_id = "Salesforce/blip-image-captioning-large"
-
-model = BlipForConditionalGeneration.from_pretrained(model_id)
-processor = BlipProcessor.from_pretrained(model_id)
+captioner = pipeline(model=model_id)
 
 def launch(input):
-    image = Image.open(input).convert('RGB')
-    inputs = processor(image, return_tensors="pt")
-    out = model.generate(**inputs)
-    return processor.decode(out[0], skip_special_tokens=True)
+    return captioner(input)
 
-iface = gr.Interface(launch, inputs="image", outputs="text")
+iface = gr.Interface(launch, inputs="pil", outputs="text")
 iface.launch()