sudhir2016 commited on
Commit
9d91775
·
1 Parent(s): e607307

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -10
app.py CHANGED
@@ -1,14 +1,5 @@
1
  import gradio as gr
2
  import pandas as pd
3
- def recognize(img):
4
- img1=img.resize((224,224))
5
- img_norm = transform_norm(img1)
6
- output = model(img_norm[None,])
7
- out=output.argmax(dim=-1).item()
8
- out1=(df.iloc[out,0])
9
- return out1
10
- demo = gr.Interface(fn=recognize, inputs='pil',outputs='text',examples= [['Banana.jpg']])
11
- demo.launch()#debug=True)
12
  import torch
13
  from PIL import Image
14
  from torchvision import transforms
@@ -22,7 +13,6 @@ transform_list = [
22
  transforms.Resize(int((256 / 224) * input_size), interpolation=InterpolationMode.BICUBIC),
23
  transforms.CenterCrop(input_size)
24
  ]
25
- #transform_vis = transforms.Compose(transform_list)
26
  transform_norm = transforms.Compose(transform_list + [
27
  transforms.ToTensor(),
28
  transforms.Normalize(IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD),
 
1
  import gradio as gr
2
  import pandas as pd
 
 
 
 
 
 
 
 
 
3
  import torch
4
  from PIL import Image
5
  from torchvision import transforms
 
13
  transforms.Resize(int((256 / 224) * input_size), interpolation=InterpolationMode.BICUBIC),
14
  transforms.CenterCrop(input_size)
15
  ]
 
16
  transform_norm = transforms.Compose(transform_list + [
17
  transforms.ToTensor(),
18
  transforms.Normalize(IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD),