Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import os
 from random import randint
 from all_models import models
 from datetime import datetime
@@ -7,6 +8,8 @@ import numpy as np
 import time
 import requests
 import logging
+
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 logging.basicConfig(level=logging.WARNING)
 now2 = 0
 index_tracker = 0  # Index tracker for the current model