yucornetto commited on
Commit
f1c062a
1 Parent(s): 506552e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,6 @@
2
  from PIL import Image
3
  import gradio as gr
4
  from imagenet_classes import imagenet_idx2classname
5
- from huggingface_hub import hf_hub_download
6
  import torch
7
  torch.backends.cuda.matmul.allow_tf32 = True
8
  torch.backends.cudnn.allow_tf32 = True
@@ -13,6 +12,7 @@ import os
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
 
 
16
  model2ckpt = {
17
  "TiTok-L-32": ("tokenizer_titok_l32.bin", "generator_titok_l32.bin"),
18
  }
 
2
  from PIL import Image
3
  import gradio as gr
4
  from imagenet_classes import imagenet_idx2classname
 
5
  import torch
6
  torch.backends.cuda.matmul.allow_tf32 = True
7
  torch.backends.cudnn.allow_tf32 = True
 
12
 
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
+
16
  model2ckpt = {
17
  "TiTok-L-32": ("tokenizer_titok_l32.bin", "generator_titok_l32.bin"),
18
  }