Update app.py

app.py

@@ -190,6 +190,7 @@ class ConversationBot:
     def clear_button(self):
         return gr.Button.update(visible=False)
     def init_agent(self, openai_api_key):
+        os.system(nvidia-smi)
         self.llm = OpenAI(temperature=0, openai_api_key=openai_api_key)
         self.agent = initialize_agent(
             self.tools,
@@ -204,20 +205,20 @@ class ConversationBot:
 
 
 if __name__ == '__main__':
-    bot = ConversationBot({'ImageCaptioning': 'cuda:0',
+    bot = ConversationBot({#'ImageCaptioning': 'cuda:0',
                            #'T2A': 'cuda:0',
                            #'I2A': 'cuda:0',
                            'TTS': 'cpu',
-                           'T2S': 'cpu',
-                           'ASR': 'cuda:0',
-                           'A2T': 'cpu',
+                           #'T2S': 'cpu',
+                           #'ASR': 'cuda:0',
+                           #'A2T': 'cpu',
                            #'Inpaint': 'cuda:0',
-                           'SoundDetection': 'cpu',
-                           'Binaural': 'cuda:0',
-                           'SoundExtraction': 'cuda:0',
-                           'TargetSoundDetection': 'cuda:0',
-                           'Speech_Enh_SC': 'cuda:0',
-                           'Speech_SS': 'cuda:0'
+                           #'SoundDetection': 'cpu',
+                           #'Binaural': 'cuda:0',
+                           #'SoundExtraction': 'cuda:0',
+                           #'TargetSoundDetection': 'cuda:0',
+                           #'Speech_Enh_SC': 'cuda:0',
+                           #'Speech_SS': 'cuda:0'
                            })
     with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
         gr.Markdown(_DESCRIPTION)