jon-fernandes commited on
Commit
c5d3347
1 Parent(s): b947cf2

Updated from colab

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. cifar10 +1 -0
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, share=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