jon-fernandes
commited on
Commit
•
c5d3347
1
Parent(s):
b947cf2
Updated from colab
Browse files
app.py
CHANGED
@@ -12,4 +12,4 @@ def classify_image(image):
|
|
12 |
pred = torch.argmax(outp.logits, dim=1).item()
|
13 |
return model.config.id2label[pred]
|
14 |
|
15 |
-
interface = gr.Interface(fn=classify_image, inputs=gr.Image(shape=(224, 224)), outputs="text").launch(debug=True
|
|
|
12 |
pred = torch.argmax(outp.logits, dim=1).item()
|
13 |
return model.config.id2label[pred]
|
14 |
|
15 |
+
interface = gr.Interface(fn=classify_image, inputs=gr.Image(shape=(224, 224)), outputs="text").launch(debug=True)
|
cifar10
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
Subproject commit b947cf2be09bbedceca2a9fa56241ce87539690d
|