shyamgupta196 commited on
Commit
098663c
1 Parent(s): 956b27f
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -22,8 +22,7 @@ def predict(img):
22
 
23
 
24
  transform = create_transform(**resolve_data_config({},model=model))
25
- # we do not need to train model , hence using model.eval() to use it only for inference
26
- model.eval()
27
 
28
  iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(type='pil'), outputs="label").launch()
29
  iface.launch()
 
22
 
23
 
24
  transform = create_transform(**resolve_data_config({},model=model))
25
+
 
26
 
27
  iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(type='pil'), outputs="label").launch()
28
  iface.launch()