RayanRen commited on
Commit
d57b721
·
1 Parent(s): 469c8ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import torch
5
  from model import create_effnetb2_model
6
  from timeit import default_timer as timer
7
  from typing import Tuple,Dict
 
8
 
9
  class_names=["pissa", "steak", "sushi"]
10
  effnetb2, effnetb2_transforms = create_effnetb2_model(num_classes=3)
 
5
  from model import create_effnetb2_model
6
  from timeit import default_timer as timer
7
  from typing import Tuple,Dict
8
+ import random
9
 
10
  class_names=["pissa", "steak", "sushi"]
11
  effnetb2, effnetb2_transforms = create_effnetb2_model(num_classes=3)