Hugging Face's logo

Spaces:
IFMedTechdemo
/
Dental_Classifier
Sleeping

App Files Community
IFMedTech commited on
Commit
aef4eda
·
verified ·
1 Parent(s): 6764cd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -5,12 +5,12 @@ import gradio as gr
5
 
6
  # Updated class names with 'plaque' in front of 'calculus' and 'gingivitis'
7
  class_names = [
8
- "plaque_calculus",
9
- "caries",
10
- "plaque_gingivitis",
11
- "hypodontia",
12
- "mouth_ulcer",
13
- "tooth_discoloration"
14
  ]
15
 
16
  # Load the model and update the final fully connected layer
 
5
 
6
  # Updated class names with 'plaque' in front of 'calculus' and 'gingivitis'
7
  class_names = [
8
+ "Plaque Calculus",
9
+ "Caries",
10
+ "Plaque Gingivitis",
11
+ "Hypodontia",
12
+ "Mouth Ulcer",
13
+ "Tooth Discoloration"
14
  ]
15
 
16
  # Load the model and update the final fully connected layer