Spaces:
Runtime error
Runtime error
Commit ·
bb65de9
1
Parent(s): d726caf
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,8 +17,8 @@ example_list=['1.jpg','2.jpg']
|
|
| 17 |
def predict(img):
|
| 18 |
# Apply Transformations
|
| 19 |
# img=np.array(img)
|
| 20 |
-
img=transform_gen(img)
|
| 21 |
-
img=img['image'].unsqueeze(0)
|
| 22 |
# Predic
|
| 23 |
gen.eval()
|
| 24 |
with torch.inference_mode():
|
|
|
|
| 17 |
def predict(img):
|
| 18 |
# Apply Transformations
|
| 19 |
# img=np.array(img)
|
| 20 |
+
img=transform_gen(img).unsqueeze(0)
|
| 21 |
+
# img=img['image'].unsqueeze(0)
|
| 22 |
# Predic
|
| 23 |
gen.eval()
|
| 24 |
with torch.inference_mode():
|