Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,8 +17,8 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
17 |
|
18 |
model_name = {
|
19 |
"aurora": 'huggan/fastgan-few-shot-aurora',
|
20 |
-
|
21 |
-
"painting":"geninhu/fastgan-few-shot-art",
|
22 |
"shell": 'huggan/fastgan-few-shot-shells',
|
23 |
"fauvism": 'huggan/fastgan-few-shot-fauvism-still-life',
|
24 |
}
|
|
|
17 |
|
18 |
model_name = {
|
19 |
"aurora": 'huggan/fastgan-few-shot-aurora',
|
20 |
+
"painting": 'huggan/fastgan-few-shot-painting',
|
21 |
+
#"painting":"geninhu/fastgan-few-shot-art",
|
22 |
"shell": 'huggan/fastgan-few-shot-shells',
|
23 |
"fauvism": 'huggan/fastgan-few-shot-fauvism-still-life',
|
24 |
}
|