Spaces:
Sleeping
Sleeping
reveseforward
commited on
Commit
·
302d604
1
Parent(s):
5817307
save
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import gradio as gr
|
|
| 5 |
# ----------------------------
|
| 6 |
# CONFIG
|
| 7 |
# ----------------------------
|
| 8 |
-
MODEL_NAME = "reverseforward/
|
| 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 |
|