Update app.py

app.py

@@ -291,13 +291,13 @@ class BLIPVQA:
 class ConversationBot:
     def __init__(self):
         print("Initializing VisualChatGPT")
-        self.edit = ImageEditing(device="cuda")
-        self.i2t = ImageCaptioning(device="cuda")
-        self.t2i = T2I(device="cuda")
+        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")
-        self.BLIPVQA = BLIPVQA(device="cuda")
-        self.pix2pix = Pix2Pix(device="cuda")
+        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,
@@ -420,12 +420,12 @@ class ConversationBot:
 
 bot = ConversationBot()
 with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as demo:
-    openai_api_key_input = gr.Textbox(type = "password", label = "Enter your OpenAI API key here")
-    chatbot = gr.Chatbot(elem_id="chatbot", label="Visual ChatGPT")
-    state = gr.State([])
-    
+    openai_api_key_input = gr.Textbox(type = "password", label = "Enter your OpenAI API key here")    
     
     with gr.Row(visible = False) as input_row:
+        chatbot = gr.Chatbot(elem_id="chatbot", label="Visual ChatGPT")
+        state = gr.State([])
+
         with gr.Column(scale=0.7):
             txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter, or upload an image").style(container=False)
         with gr.Column(scale=0.15, min_width=0):