diff --git a/main.py b/main.py index fe706cf..6943169 100644 --- a/main.py +++ b/main.py @@ -4,8 +4,6 @@ import os from util import download_weights os.environ["GRADIO_ANALYTICS_ENABLED"] = "False" def run(): - if not torch.cuda.is_available(): - print("Warning: GPU is not available, we will use CPU, the application may run slower!\nyou computer will very likely heat up!") download_weights.download() app.run()