Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@
|
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
import torch
|
|
|
5 |
|
6 |
from timeit import default_timer as timer
|
7 |
from model import create_effnetb2_model
|
@@ -110,6 +111,8 @@ class_names = ['apple_pie',
|
|
110 |
'tuna_tartare',
|
111 |
'waffles']
|
112 |
|
|
|
|
|
113 |
from torchvision.models._api import WeightsEnum
|
114 |
from torch.hub import load_state_dict_from_url
|
115 |
def get_state_dict(self, *args, **kwargs):
|
|
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
import torch
|
5 |
+
import numpy
|
6 |
|
7 |
from timeit import default_timer as timer
|
8 |
from model import create_effnetb2_model
|
|
|
111 |
'tuna_tartare',
|
112 |
'waffles']
|
113 |
|
114 |
+
|
115 |
+
print(numpy.__version__)
|
116 |
from torchvision.models._api import WeightsEnum
|
117 |
from torch.hub import load_state_dict_from_url
|
118 |
def get_state_dict(self, *args, **kwargs):
|