Sandhya commited on
Commit
9f77617
·
1 Parent(s): e1d34a8

change app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  import os
3
  import torch
4
  from model import create_model
5
- from timeit import default_timer as Timer
6
  class_names=["Pizza","Steak","Sushi"]
7
  effnetb2,effnetb2_transform=create_model(num_class=3)
8
  effnetb2.load_state_dict(torch.load(f="effnetb2_pizza_steak_sushi (1).pt",map_location=torch.device('cpu')))
 
2
  import os
3
  import torch
4
  from model import create_model
5
+ from timeit import default_timer as timer
6
  class_names=["Pizza","Steak","Sushi"]
7
  effnetb2,effnetb2_transform=create_model(num_class=3)
8
  effnetb2.load_state_dict(torch.load(f="effnetb2_pizza_steak_sushi (1).pt",map_location=torch.device('cpu')))