Update app.py

app.py

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