geninhu commited on
Commit
59e8e99
·
1 Parent(s): 20a5a58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,7 +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
  "shell": 'huggan/fastgan-few-shot-shells',
22
  "fauvism": 'huggan/fastgan-few-shot-fauvism-still-life',
23
  }
 
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
  }