Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from torchvision import transforms
|
|
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():
|
|
|
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 |
+
#zheshizhushi
|
10 |
def predict(inp):
|
11 |
inp = transforms.ToTensor()(inp).unsqueeze(0)
|
12 |
with torch.no_grad():
|