Spaces:
Sleeping
Sleeping
Saurav Chaudhari commited on
Commit ·
934926c
1
Parent(s): 7cd4453
Add application file
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ def analyze_image(input_img):
|
|
| 22 |
|
| 23 |
np_img = np.array(input_img)
|
| 24 |
# Standardize input for pyiqa
|
| 25 |
-
img_tensor = pyiqa.utils.img2tensor(
|
| 26 |
|
| 27 |
results = []
|
| 28 |
with torch.no_grad():
|
|
|
|
| 22 |
|
| 23 |
np_img = np.array(input_img)
|
| 24 |
# Standardize input for pyiqa
|
| 25 |
+
img_tensor = pyiqa.utils.img2tensor(np_img).unsqueeze(0).to(device)
|
| 26 |
|
| 27 |
results = []
|
| 28 |
with torch.no_grad():
|