Update app.py

app.py

@@ -291,28 +291,14 @@ class BLIPVQA:
 class ConversationBot:
     def __init__(self):
         print("Initializing VisualChatGPT")
-        self.llm = OpenAI(temperature=0)
-        self.edit = ImageEditing(device="cuda:6")
-        self.i2t = ImageCaptioning(device="cuda:4")
+        self.llm = OpenAI(temperature=0, openai_api_key = )
+        self.edit = ImageEditing(device="cuda:1")
+        self.i2t = ImageCaptioning(device="cuda:1")
         self.t2i = T2I(device="cuda:1")
         self.image2canny = image2canny()
         self.canny2image = canny2image(device="cuda:1")
-        self.image2line = image2line()
-        self.line2image = line2image(device="cuda:1")
-        self.image2hed = image2hed()
-        self.hed2image = hed2image(device="cuda:2")
-        self.image2scribble = image2scribble()
-        self.scribble2image = scribble2image(device="cuda:3")
-        self.image2pose = image2pose()
-        self.pose2image = pose2image(device="cuda:3")
-        self.BLIPVQA = BLIPVQA(device="cuda:4")
-        self.image2seg = image2seg()
-        self.seg2image = seg2image(device="cuda:7")
-        self.image2depth = image2depth()
-        self.depth2image = depth2image(device="cuda:7")
-        self.image2normal = image2normal()
-        self.normal2image = normal2image(device="cuda:5")
-        self.pix2pix = Pix2Pix(device="cuda:3")
+        self.BLIPVQA = BLIPVQA(device="cuda:1")
+        self.pix2pix = Pix2Pix(device="cuda:1")
         self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='output')
         self.tools = [
             Tool(name="Get Photo Description", func=self.i2t.inference,