Update app.py
Browse files
app.py
CHANGED
|
@@ -45,6 +45,7 @@ if uploaded_file is not None:
|
|
| 45 |
tensor = cust_transform(image=img)
|
| 46 |
tensor = tensor['image'].float().resize(1,3,256,256)
|
| 47 |
|
|
|
|
| 48 |
custom_pred = model.forward(tensor).detach().numpy()
|
| 49 |
custom_pred
|
| 50 |
id2class[np.argmax(custom_pred)]
|
|
|
|
| 45 |
tensor = cust_transform(image=img)
|
| 46 |
tensor = tensor['image'].float().resize(1,3,256,256)
|
| 47 |
|
| 48 |
+
model.eval()
|
| 49 |
custom_pred = model.forward(tensor).detach().numpy()
|
| 50 |
custom_pred
|
| 51 |
id2class[np.argmax(custom_pred)]
|