Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from typing import Tuple,Dict
|
|
| 9 |
|
| 10 |
class_name=["pizza","steak","sushi"]
|
| 11 |
|
| 12 |
-
effnetb3,
|
| 13 |
|
| 14 |
effnetb3.load_state_dict(
|
| 15 |
torch.load(
|
|
@@ -22,7 +22,7 @@ def predict(img) -> Tuple[Dict,float]:
|
|
| 22 |
|
| 23 |
start_time=Timer()
|
| 24 |
|
| 25 |
-
img=
|
| 26 |
|
| 27 |
effnetb3.eval()
|
| 28 |
with torch.inference_mode():
|
|
|
|
| 9 |
|
| 10 |
class_name=["pizza","steak","sushi"]
|
| 11 |
|
| 12 |
+
effnetb3,effentb3_tranforms=model_efficientb3(out_feature=3)
|
| 13 |
|
| 14 |
effnetb3.load_state_dict(
|
| 15 |
torch.load(
|
|
|
|
| 22 |
|
| 23 |
start_time=Timer()
|
| 24 |
|
| 25 |
+
img=effentb3_tranforms(img).unsqueeze(0)
|
| 26 |
|
| 27 |
effnetb3.eval()
|
| 28 |
with torch.inference_mode():
|