add openai

app.py

@@ -132,7 +132,19 @@ def process_image(
         image=image_input,
         task='<OD>'
     )
-    caption = result
+    from openai import OpenAI
+    client = OpenAI()
+
+    completion = client.chat.completions.create(
+    model="gpt-4o",
+    messages=[
+        {"role": "system", "content": "You are a helpful assistant."},
+        {"role": "user", "content": "Hello!"}
+    ]
+    )
+
+    print(completion.choices[0].message.content)
+    caption = ', '.join(result['<OD>']['labels'])
     _, result = run_florence_inference(
         model=FLORENCE_MODEL,
         processor=FLORENCE_PROCESSOR,

requirements.txt

@@ -7,4 +7,5 @@ samv2
 gradio
 supervision
 opencv-python
-pytest
+pytest
+openai