Update app.py

app.py

@@ -229,6 +229,8 @@ def predict(
     generator: A generator that yields the chatbot outputs, history, and status.
     """
     print("running the prediction function")
+    import os
+    os.system('nvidia-smi')
     try:
         tokenizer, vl_gpt, vl_chat_processor = models[model_select_dropdown]