geninhu commited on
Commit
4e206a5
1 Parent(s): 59e8e99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- #"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
  }
 
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
  }