Spaces:
Sleeping
Sleeping
Sandhya
commited on
Commit
·
9f77617
1
Parent(s):
e1d34a8
change app
Browse files
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
|
| 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')))
|