Update app.py

app.py

@@ -4,7 +4,7 @@ import base64
 import gradio as gr
 
 from dotenv import load_dotenv
-from openai import OpenAI
+from openai import AzureOpenAI
 
 
 def generate_data_uri(png_file_path):
@@ -21,7 +21,7 @@ def generate_data_uri(png_file_path):
     return data_uri
 
 
-def decision(png_file_path, client: OpenAI, lmm: str) -> str:
+def decision(png_file_path, client, lmm: str) -> str:
 
     image_data = generate_data_uri(png_file_path)
 
@@ -71,20 +71,19 @@ def decision(png_file_path, client: OpenAI, lmm: str) -> str:
 
     return decision
 
-def predict(image, anyscale_api_key = os.environ['ANYSCALE_API_KEY']):
+def predict(image):
 
-    lmm = "llava-hf/llava-v1.6-mistral-7b-hf"
+    lmm = "gpt-4o"
 
-    client = OpenAI(
-        base_url="https://api.endpoints.anyscale.com/v1",
-        api_key=anyscale_api_key
+    client = AzureOpenAI(
+        api_key = os.environ["AZURE_OPENAI_KEY"],
+        azure_endpoint = os.environ["AZURE_OPENAI_ENDPOINT"],
+        api_version = "2024-02-01"
     )
     
     verdict = decision(image, client, lmm)
     return verdict
 
-load_dotenv()
-
 demo = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="filepath", label="Upload your image"),