From 37ac7f4262c9e18d9e83b9e044a8a38728204b60 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Fri, 6 Oct 2023 01:04:28 -0400
Subject: [PATCH] clean up

---
 apps/omni_ui.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/apps/omni_ui.py b/apps/omni_ui.py
index 13c76dda..05ac8194 100644
--- a/apps/omni_ui.py
+++ b/apps/omni_ui.py
@@ -4,8 +4,8 @@ import threading
 import os
 import glob
 import base64
-from langchain.llms import ChatOpenAI  #
-from swarms.agents import OmniModalAgent  # Replace with your actual class
+from langchain.llms import OpenAIChat  
+from swarms.agents import OmniModalAgent  
 
 #Function to convert image to base64
 def image_to_base64(image_path):
@@ -21,7 +21,7 @@ def get_latest_image():
     return latest_file
 
 #Initialize your OmniModalAgent
-llm = ChatOpenAI(model_name="gpt-4")  # Replace with your actual initialization
+llm = OpenAIChat(model_name="gpt-4")  # Replace with your actual initialization
 agent = OmniModalAgent(llm)  # Replace with your actual initialization
 
 #Global variable to store chat history