ranjith commited on
Commit
729b213
1 Parent(s): 5890f7a

description added

Browse files
Files changed (9) hide show
  1. app.py +6 -3
  2. basalt.jpeg +0 -0
  3. basalt1.jpeg +0 -0
  4. limestone.jpeg +0 -0
  5. limestone1.jpg +0 -0
  6. marble.jpeg +0 -0
  7. marble1.jpeg +0 -0
  8. sandstone.jpeg +0 -0
  9. sandstone4.jpeg +0 -0
app.py CHANGED
@@ -13,6 +13,9 @@ def predict(img):
13
  pred,pred_idx,probs = learn.predict(img)
14
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
15
 
16
- title = "rock classifier"
17
- examples = ["basalt1.jpeg","coal.jpeg","granite.jpeg", "limestone.jpeg","marble.jpeg","quartzite.jpeg","sandstone.jpeg"]
18
- gr.Interface(fn=predict, inputs=gr.Image(), outputs=gr.Label(num_top_classes=7),title = title,examples= examples).launch(share=True)
 
 
 
 
13
  pred,pred_idx,probs = learn.predict(img)
14
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
15
 
16
+ title = "<div style='text-align:center'> Rock classifier</div>"
17
+ examples = ["basalt.jpeg","coal.jpeg","granite.jpeg", "limestone1.jpeg","marble1.jpeg","quartzite.jpeg","sandstone4.jpeg"]
18
+ description = "Rock classification using convolutional neural networks results may not be accurate since trained on small dataset"
19
+ article="<p style='text-align: center'><a href='https://www.linkedin.com/in/amrutha-kp/' target='_blank'>Linkedin</a></p>"
20
+
21
+ gr.Interface(fn=predict, inputs=gr.Image(), outputs=gr.Label(num_top_classes=7),title = title,examples= examples,article=article,description=description).launch(share=True)
basalt.jpeg ADDED
basalt1.jpeg DELETED
Binary file (11.9 kB)
 
limestone.jpeg DELETED
Binary file (5.43 kB)
 
limestone1.jpg ADDED
marble.jpeg DELETED
Binary file (6.58 kB)
 
marble1.jpeg ADDED
sandstone.jpeg DELETED
Binary file (8.48 kB)
 
sandstone4.jpeg ADDED