Spaces:
Runtime error
Runtime error
AnishaNaik03 commited on
Commit ·
13270d0
1
Parent(s): 1bc65a2
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
|