Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -229,6 +229,8 @@ def predict(
|
|
229 |
generator: A generator that yields the chatbot outputs, history, and status.
|
230 |
"""
|
231 |
print("running the prediction function")
|
|
|
|
|
232 |
try:
|
233 |
tokenizer, vl_gpt, vl_chat_processor = models[model_select_dropdown]
|
234 |
|
|
|
229 |
generator: A generator that yields the chatbot outputs, history, and status.
|
230 |
"""
|
231 |
print("running the prediction function")
|
232 |
+
import os
|
233 |
+
os.system('nvidia-smi')
|
234 |
try:
|
235 |
tokenizer, vl_gpt, vl_chat_processor = models[model_select_dropdown]
|
236 |
|