Spaces:
Sleeping
Sleeping
AnishaNaik03 commited on
Commit ·
f5cc65d
1
Parent(s): 5f5d766
initial commit
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -191,7 +191,7 @@ model_path = r"topmodel.pkl"
|
|
| 191 |
|
| 192 |
with open(model_path, "rb") as f:
|
| 193 |
G = pickle.load(f)["G_ema"]
|
| 194 |
-
G.eval().to("
|
| 195 |
|
| 196 |
|
| 197 |
|
|
|
|
| 191 |
|
| 192 |
with open(model_path, "rb") as f:
|
| 193 |
G = pickle.load(f)["G_ema"]
|
| 194 |
+
G.eval().to("cpu")
|
| 195 |
|
| 196 |
|
| 197 |
|
requirements.txt
CHANGED
|
@@ -6,3 +6,4 @@ scikit-learn>=0.24.0
|
|
| 6 |
gradio
|
| 7 |
torch==2.1.0
|
| 8 |
numpy<2
|
|
|
|
|
|
| 6 |
gradio
|
| 7 |
torch==2.1.0
|
| 8 |
numpy<2
|
| 9 |
+
ninja
|