Spaces:
Runtime error
Runtime error
Commit ·
5cc0148
1
Parent(s): 25cc5fe
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,7 +17,7 @@ example_list=['1.jpg','2.jpg']
|
|
| 17 |
def predict(img):
|
| 18 |
# Apply Transformations
|
| 19 |
img=np.array(img)
|
| 20 |
-
img=transform_gen(
|
| 21 |
img=img['image'].unsqueeze(0)
|
| 22 |
# Predic
|
| 23 |
gen.eval()
|
|
|
|
| 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()
|