Hugging Face's logo

Spaces:
eulermaxwell
/
AnalogRetriever
Sleeping

App Files Community
eulermax commited on
Commit
db636db
·
verified ·
1 Parent(s): f4b57d6

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from PIL import Image
15
  from huggingface_hub import hf_hub_download, snapshot_download
16
  import requests
17
 
18
- MODEL_REPO = "eulermax/AnalogRetriever-model"
19
  DATA_REPO = "eulermaxwell/AnalogRetriever-data"
20
  CKPT_NAME = "tic_cur_v2_best_ablation_1k.pth"
21
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
15
  from huggingface_hub import hf_hub_download, snapshot_download
16
  import requests
17
 
18
+ MODEL_REPO = "eulermaxwell/AnalogRetriever-model"
19
  DATA_REPO = "eulermaxwell/AnalogRetriever-data"
20
  CKPT_NAME = "tic_cur_v2_best_ablation_1k.pth"
21
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")