Hugging Face's logo

Spaces:
Pravalika56
/
resnetimageclassification
Sleeping

App Files Community
Pravalika56 commited on
Commit
3ea5405
·
verified ·
1 Parent(s): 7ac5cf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import requests
3
  import gradio as gr
4
  from PIL import Image
5
 
6
- model = torch.hub.load("pytorch/vision/v0.17", "resnet18", pre_trained = True).eval()
7
  response = request.get("https://git.io/JJkYN")
8
  labels = response.text.split("\n")
9
 
 
3
  import gradio as gr
4
  from PIL import Image
5
 
6
+ model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained = True).eval()
7
  response = request.get("https://git.io/JJkYN")
8
  labels = response.text.split("\n")
9