mohamedsaeed823 commited on
Commit
a7c81c7
1 Parent(s): 69070c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,6 +1,8 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
 
 
4
  def is_metal_album(x): return x[0].isupper()
5
 
6
  learn = load_learner('model.pkl')
@@ -16,4 +18,4 @@ def predict(img):
16
 
17
  examples = ['metalkinggizz.png', 'popkinggizz.jpeg', 'TI.png','TI2.jpeg']
18
 
19
- gr.Interface(fn=predict, inputs=gr.Image(height=512,width=512), outputs=gr.Label(num_top_classes=2),examples=examples).launch(share=True)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
+ description = 'this is a model that predicts whether an album is metal or pop based on its cover art'
5
+
6
  def is_metal_album(x): return x[0].isupper()
7
 
8
  learn = load_learner('model.pkl')
 
18
 
19
  examples = ['metalkinggizz.png', 'popkinggizz.jpeg', 'TI.png','TI2.jpeg']
20
 
21
+ gr.Interface(fn=predict, inputs=gr.Image(height=512,width=512), outputs=gr.Label(num_top_classes=2), examples=examples, description=description).launch(share=True)