mirror of
https://github.com/yuruotong1/autoMate.git
synced 2025-12-26 05:16:21 +08:00
14 lines
430 B
Python
14 lines
430 B
Python
from gradio_ui import app
|
|
from util import download_weights
|
|
import torch
|
|
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!")
|
|
print("Downloading the weight files...")
|
|
# download the weight files
|
|
download_weights.download_models()
|
|
app.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run() |