Create app.py

app.py

@@ -0,0 +1,24 @@
+import spaces
+import gradio as gr
+from transformers import BlipForConditionalGeneration
+
+model = BlipForConditionalGeneration.from_pretrained(
+    "Salesforce/blip-image-captioning-base")
+model.to("cuda")
+
+processor = AutoProcessor.from_pretrained(
+    "Salesforce/blip-image-captioning-base")
+
+
+@spaces.GPU
+def image_caption(input):
+    inputs = processor(input, return_tensors="pt")
+    out = model.generate(**inputs)
+    return processor.decode(out[0], skip_special_tokens = True)
+
+
+iface = gr.Interface(fn = image_caption,
+                     inputs = gr.Image(type='pil'),
+                     outputs = "text")
+
+iface.launch()