initialize

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+import open_clip
+
+
+model, _, transform = open_clip.create_model_and_transforms(
+    "coca_ViT-B-32",
+    pretrained="laion2B-s13B-b90k"
+)
+
+def output_generate(image):
+    im = transform(image).unsqueeze(0)
+    generated = model.generate(im)
+    return open_clip.decode(generated).split("<end_of_text>")[0].replace("<start_of_text>", "")
+
+iface = gr.Interface(fn=output_generate, inputs=gr.image(type="pil"), outputs="text")
+iface.launch()

requirements.txt

@@ -0,0 +1 @@
+open_clip_torch