Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ MODEL_IDS = {
|
|
| 20 |
EXAMPLE_IMAGES = {
|
| 21 |
"去雨痕 (Derain)": [["derain_example1.png"], ["derain_example2.png"], ["derain_example3.png"]],
|
| 22 |
"去雨滴 (Deraindrop)": [["deraindrop_example1.png"], ["deraindrop_example2.png"], ["deraindrop_example3.png"]],
|
| 23 |
-
"去雾 (Dehaze)": []
|
| 24 |
}
|
| 25 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 26 |
print(f"正在使用的设备: {device}")
|
|
|
|
| 20 |
EXAMPLE_IMAGES = {
|
| 21 |
"去雨痕 (Derain)": [["derain_example1.png"], ["derain_example2.png"], ["derain_example3.png"]],
|
| 22 |
"去雨滴 (Deraindrop)": [["deraindrop_example1.png"], ["deraindrop_example2.png"], ["deraindrop_example3.png"]],
|
| 23 |
+
"去雾 (Dehaze)": [["dehaze_example1.jpg"],["dehaze_example2.jpg"],["dehaze_example3.jpg"]]
|
| 24 |
}
|
| 25 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 26 |
print(f"正在使用的设备: {device}")
|