Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import base64
 
-from langchain.callbacks import get_openai_callback
 from openai import OpenAI
 
 config = {
@@ -23,22 +22,20 @@ def invoke(openai_api_key, prompt, image):
         raise gr.Error("Image is required.")
 
     content = ""
-    cb = ""
     
     try:
         client = OpenAI(api_key = openai_api_key)
 
         img_b64 = get_img_b64(image)
 
-        with get_openai_callback() as cb:
-            completion = client.chat.completions.create(
-                max_tokens = config["max_tokens"],
-                messages=[{"role": "user",
-                           "content": [{"type": "text", "text": prompt},
-                                       {"type": "image_url",
-                                        "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"},},],},],
-                model = config["model"],
-                temperature = config["temperature"],)
+        completion = client.chat.completions.create(
+            max_tokens = config["max_tokens"],
+            messages=[{"role": "user",
+                       "content": [{"type": "text", "text": prompt},
+                                   {"type": "image_url",
+                                    "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"},},],},],
+            model = config["model"],
+            temperature = config["temperature"],)
     
         content = completion.choices[0].message.content
     except Exception as e:
@@ -46,7 +43,7 @@ def invoke(openai_api_key, prompt, image):
 
         raise gr.Error(e)
 
-    return content, cb
+    return content, ""
 
 description = """<a href='https://www.gradio.app/'>Gradio</a> UI using the <a href='https://openai.com/'>OpenAI</a> API 
                  with <a href='https://openai.com/research/gpt-4'>gpt-4-vision-preview</a> model."""
@@ -62,4 +59,4 @@ demo = gr.Interface(fn = invoke,
                     title = "Generative AI - Multimodal LLM",
                     description = description,)
 
-demo.launch()
+demo.launch()