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 |
}
|