Update app.py

app.py

@@ -1,5 +1,19 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.Interface.load("models/ronniet/git-base-env").launch()
+captioner = pipeline(model="microsoft/git-base")
+
+def predict(image):
+    return captioner(image)[0]["generated_text"]
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image,
+    outputs='text'
+)
+
+demo.launch()
+
+# gr.Interface.load("models/ronniet/git-base-env").launch()
 
 # gr.Interface.load("models/microsoft/git-base").launch()