Update app.py

app.py

@@ -1,31 +1,15 @@
-
-from transformers import pipeline
-from transformers.utils import logging
 import os
-import gradio as gr
-from PIL import Image
-# Use a pipeline as a high-level helper
 from transformers import pipeline
 
+pipe = pipeline("image-to-text",
+                model="./models/Salesforce/blip-image-captioning-base")
 
+def launch(input):
+    out = pipe(input)
+    return out[0]['generated_text']
 
+app = gr.Interface(launch,
+                     inputs=gr.Image(type='pil'),
+                     outputs="text")
 
-od_pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
-
-def get_pipeline_prediction(pil_image):
-    
-    pipeline_output = od_pipe(pil_image)
-    
-    processed_image = render_results_in_image(pil_image,
-                                            pipeline_output)
-    return processed_image
-
-demo = gr.Interface(
-  fn=get_pipeline_prediction,
-  inputs=gr.Image(label="Input image", 
-                  type="pil"),
-  outputs=gr.Image(label="Output image with predicted instances",
-                   type="pil")
-)
-
-demo.launch(share=False)
+app.launch()