drkareemkamal commited on
Commit
abed5b0
·
verified ·
1 Parent(s): f7b2cea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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):