Update demo.py

demo.py

@@ -8,11 +8,14 @@ import gradio as gr
 
 os.system("python tools/download_models.py") # -> dummy extra
 
-if warning == 'True':
+
+
+
+if config.iscolab:
     warning = True
-elif warning == 'False':
-    warning = False
 
+else:
+    Warning = False
 
 
 # gr.themes.builder()