Update app.py

app.py

@@ -24,6 +24,7 @@ from io import BytesIO
 import spaces
 import gradio as gr
 import torch
+import os
 from app_modules.gradio_utils import (
     cancel_outputing,
     delete_last_conversation,
@@ -228,6 +229,7 @@ def predict(
     generator: A generator that yields the chatbot outputs, history, and status.
     """
     print("running the prediction function")
+    os.system('nvidia-smi')
     try:
         tokenizer, vl_gpt, vl_chat_processor = models[model_select_dropdown]