Hugging Face's logo

Spaces:
AlexWortega
/
food_calories
Build error

App Files Community
AlexWortega commited on
Commit
16b1928
·
1 Parent(s): 9e05fcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ import gradio as gr
11
  from rudolph import utils
12
  device = 'cuda'
13
 
14
- model = get_rudolph_model('350M', fp16=True, device='cuda')
 
15
  tokenizer = get_tokenizer()
16
  vae = get_vae(dwt=False).to(device)
17
 
 
11
  from rudolph import utils
12
  device = 'cuda'
13
 
14
+ model = get_rudolph_model('350M', fp16=True, device=device)
15
+ model.load_state_dict(torch.load("awesomemodel__dalle_1500.pt"))
16
  tokenizer = get_tokenizer()
17
  vae = get_vae(dwt=False).to(device)
18