Update app.py

app.py

@@ -1,36 +1,26 @@
-import gradio as gr
+import torch
+
+model = torch.hub.load('pytorch/vision:v0.6.0', 'resnet18', pretrained=True).eval()
+
+import requests
+from torchvision import transforms
 
-def calculator(num1, operation, num2, operation, num3):
-    if operation == "add":
-        return num1 + num2 + num3
-    elif operation == "subtract":
-        return num1 - num2 - num3
-    elif operation == "multiply":
-        return num1 * num2 * num3
-    elif operation == "divide":
-        if num2 == 0:
-            raise gr.Error("Cannot divide by zero!")
-        return num1 / num2 / num3
+# Download human-readable labels for ImageNet.
+response = requests.get("https://git.io/JJkYN")
+labels = response.text.split("\n")
+
+def predict(inp):
+  inp = transforms.ToTensor()(inp).unsqueeze(0)
+  with torch.no_grad():
+    prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
+    confidences = {labels[i]: float(prediction[i]) for i in range(1000)}    
+  return confidences
+
+import gradio as gr
 
-demo = gr.Interface(
-    calculator,
-    [
-        "number", 
-        gr.Radio(["add", "subtract", "multiply", "divide"]),
-        "number"
-    ],
-    "number",
-    examples=[
-        [5, "add", 3],
-        [4, "divide", 2],
-        [-4, "multiply", 2.5],
-        [0, "subtract", 1.2],
-        [508584894.9485847444, "add", 3000000000.88475775775884],
-        [4, "divide", 2],
-        [-4.585857466363363, "multiply", 6452522522.54754662525646],
-        [0, "subtract", 1.2, "divide", 56635356366.74664],
-    ],
-    title="Toy Calculator",
-    description="Here's a sample toy calculator. Allows you to calculate things like $2+2=4$",
-)
-demo.launch()
+gr.Interface(fn=predict, 
+             inputs=gr.inputs.Image(type="pil"),
+             outputs=gr.outputs.Label(num_top_classes=3),
+             examples=["lion.jpg", "cheetah.jpg"],
+             theme="default",
+             css=".footer{display:none !important}").launch()