mohammed-aljafry commited on
Commit
08cd1fb
·
verified ·
1 Parent(s): 5a7e9bd

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ MODELS_ON_HUB = {
32
  "My Interfuser Model (from Hub)": {
33
  "repo_id": "BaseerAI/Interfuser-Baseer-v1",
34
  "filename": "best_model.pth",
35
- "config_key": "interfuser_baseline"
36
  }
37
  }
38
 
 
32
  "My Interfuser Model (from Hub)": {
33
  "repo_id": "BaseerAI/Interfuser-Baseer-v1",
34
  "filename": "best_model.pth",
35
+ "config_key": "interfuser_lightweight"
36
  }
37
  }
38