Finish

app.py

@@ -3,8 +3,17 @@ import requests
 import gradio as gr
 
 from transformers import BlipProcessor, BlipForConditionalGeneration
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+model_id = "Salesforce/blip-image-captioning-base"
+
+model = BlipForConditionalGeneration.from_pretrained(model_id)
+processor = BlipProcessor.from_pretrained(model_id)
+
+def launch(input):
+    image = Image.open(requests.get(input, stream=True).raw).convert('RGB')
+    inputs = processor(image, return_tensors="pt")
+    out = model.generate(**inputs)
+    return processor.decode(out[0], skip_special_tokens=True)
+
+iface = gr.Interface(launch, inputs="text", outputs="text")
 iface.launch()

flagged/log.csv

@@ -0,0 +1,2 @@
+name,output,flag,username,timestamp
+,Hello !!,,,2023-08-18 15:35:35.677237

requirements.txt

@@ -35,7 +35,9 @@ MarkupSafe==2.1.3
 matplotlib==3.7.2
 mdit-py-plugins==0.3.3
 mdurl==0.1.2
+mpmath==1.3.0
 multidict==6.0.4
+networkx==3.1
 numpy==1.25.2
 orjson==3.9.5
 packaging==23.1
@@ -58,8 +60,12 @@ semantic-version==2.10.0
 six==1.16.0
 sniffio==1.3.0
 starlette==0.27.0
+sympy==1.12
 tokenizers==0.13.3
 toolz==0.12.0
+torch==2.0.1
+torchaudio==2.0.2
+torchvision==0.15.2
 tqdm==4.66.1
 transformers==4.31.0
 typing_extensions==4.7.1