aliabd HF staff commited on
Commit
2c1bb0d
1 Parent(s): f02201a

Delete app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +0 -23
app.py DELETED
@@ -1,23 +0,0 @@
1
- import gradio as gr
2
- import torch
3
- import requests
4
- from torchvision import transforms
5
-
6
- model = torch.hub.load('pytorch/vision:v0.6.0', 'resnet18', pretrained=True).eval()
7
- response = requests.get("https://git.io/JJkYN")
8
- labels = response.text.split("\n")
9
-
10
- def predict(inp):
11
- inp = transforms.ToTensor()(inp).unsqueeze(0)
12
- with torch.no_grad():
13
- prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
14
- confidences = {labels[i]: float(prediction[i]) for i in range(1000)}
15
- return confidences
16
-
17
- demo = gr.Interface(fn=predict,
18
- inputs=gr.inputs.Image(type="pil"),
19
- outputs=gr.outputs.Label(num_top_classes=3),
20
- examples=[["cheetah.jpg"]],
21
- )
22
-
23
- demo.launch()