Update app.py

app.py

@@ -37,18 +37,22 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent 
     try:
-        if PROVIDER_TYPE=="huggingface":
+        if PROVIDER_TYPE == "huggingface":
             llm = HuggingFaceEndpoint(
                 repo_id=REPO_ID,
                 huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
             )
             chat = ChatHuggingFace(llm=llm, verbose=True)
-        elif PROVIDER_TYPE=="openai":
+        elif PROVIDER_TYPE == "openai":
             chat = ChatOpenAI(model="gpt-4o")
         else:
             print(f"Provider {PROVIDER_TYPE} not supported.")
             return f"Provider {PROVIDER_TYPE} not supported", None
-        agent = BasicAgent(chat)
+        if include_vision == True:
+            vision_llm = ChatOpenAI(model="gpt-4o")
+        else:
+            vision_llm = None
+        agent = BasicAgent(chat, vision_llm)
 
     except Exception as e:
         print(f"Error instantiating agent: {e}")