reveseforward commited on
Commit
302d604
·
1 Parent(s): 5817307
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  # ----------------------------
6
  # CONFIG
7
  # ----------------------------
8
- MODEL_NAME = "reverseforward/qwenmeasurement" # change this to your repo name
9
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
10
  DTYPE = torch.float16 # use float16 on A10G
11
 
 
5
  # ----------------------------
6
  # CONFIG
7
  # ----------------------------
8
+ MODEL_NAME = "reverseforward/inferencemodel" # change this to your repo name
9
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
10
  DTYPE = torch.float16 # use float16 on A10G
11