Update app.py

app.py

@@ -292,13 +292,13 @@ class ConversationBot:
     def __init__(self):
         print("Initializing VisualChatGPT")
         self.llm = OpenAI(temperature=0)
-        self.edit = ImageEditing(device="cuda:1")
-        self.i2t = ImageCaptioning(device="cuda:1")
-        self.t2i = T2I(device="cuda:1")
+        self.edit = ImageEditing(device="cuda")
+        self.i2t = ImageCaptioning(device="cuda")
+        self.t2i = T2I(device="cuda")
         self.image2canny = image2canny()
-        self.canny2image = canny2image(device="cuda:1")
-        self.BLIPVQA = BLIPVQA(device="cuda:1")
-        self.pix2pix = Pix2Pix(device="cuda:1")
+        self.canny2image = canny2image(device="cuda")
+        self.BLIPVQA = BLIPVQA(device="cuda")
+        self.pix2pix = Pix2Pix(device="cuda")
         self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='output')
         self.tools = [
             Tool(name="Get Photo Description", func=self.i2t.inference,