Spaces:
Runtime error
Runtime error
Ved Gupta
commited on
Commit
·
8dd0683
1
Parent(s):
3fa04a7
Update dependencies and models
Browse files- main.py +2 -2
- requirements.txt +1 -1
main.py
CHANGED
|
@@ -28,8 +28,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 28 |
model_path_1 = "model/ImageColorizationModel.pth"
|
| 29 |
model_path_2 = "model/ImageColorizationModel-ver2.pth"
|
| 30 |
|
| 31 |
-
MODEL_VER_1 = "https://drive.google.com/uc?id=
|
| 32 |
-
MODEL_VER_2 = "https://drive.google.com/uc?id=
|
| 33 |
|
| 34 |
|
| 35 |
model = None
|
|
|
|
| 28 |
model_path_1 = "model/ImageColorizationModel.pth"
|
| 29 |
model_path_2 = "model/ImageColorizationModel-ver2.pth"
|
| 30 |
|
| 31 |
+
MODEL_VER_1 = "https://drive.google.com/uc?id=1yQzTSu6zQskzmWGDRXQGz8AgswxyJFJa"
|
| 32 |
+
MODEL_VER_2 = "https://drive.google.com/uc?id=12TLumE_HoqwrMzNq9Qu9jSPhlbn5mob9"
|
| 33 |
|
| 34 |
|
| 35 |
model = None
|
requirements.txt
CHANGED
|
@@ -26,7 +26,7 @@ filelock==3.12.0
|
|
| 26 |
fonttools==4.39.4
|
| 27 |
frozenlist==1.3.3
|
| 28 |
fsspec==2023.5.0
|
| 29 |
-
gdown==
|
| 30 |
gradio==3.30.0
|
| 31 |
gradio_client==0.2.4
|
| 32 |
h11==0.14.0
|
|
|
|
| 26 |
fonttools==4.39.4
|
| 27 |
frozenlist==1.3.3
|
| 28 |
fsspec==2023.5.0
|
| 29 |
+
gdown==5.1.0
|
| 30 |
gradio==3.30.0
|
| 31 |
gradio_client==0.2.4
|
| 32 |
h11==0.14.0
|