Anithprakash commited on
Commit
24349a5
·
verified ·
1 Parent(s): 4de048a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ from typing import Tuple,Dict
9
 
10
  class_name=["pizza","steak","sushi"]
11
 
12
- effnetb3,effentb3_tranform=model_efficientb3(out_feature=3)
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=effnetb3_tranform(img).unsqueeze(0)
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():