Spaces:

stshanks
/

scripts

Running

stshanks commited on Mar 6, 2025

Commit

71a90e9

verified ·

1 Parent(s): 095d9f1

Update app.py

updated class name order

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,8 @@ model = tf.keras.models.load_model("prescription_classification_model.keras")
 # Define the 78 drug names in the exact order corresponding to your model's output classes
 CLASS_NAMES = [
-    "Aceta", "Ace", "Alatrol", "Amodis", "Atrizin", "Axodin", "Azithrocin", "Azyth",
-    "Az", "Bacaid", "Backtone", "Baclofen", "Baclon", "Bacmax", "Beklo", "Bicozin",
     "Canazole", "Candinil", "Cetisoft", "Conaz", "Dancel", "Denixil", "Diflu",
     "Dinafex", "Disopan", "Esonix", "Esoral", "Etizin", "Exium", "Fenadin",
     "Fexofast", "Fexo", "Filmet", "Fixal", "Flamyd", "Flexibac", "Flexilax",

 # Define the 78 drug names in the exact order corresponding to your model's output classes
 CLASS_NAMES = [
+    "Ace", "Aceta", "Alatrol", "Amodis", "Atrizin", "Axodin", "Az","Azithrocin", "Azyth",
+     "Bacaid", "Backtone", "Baclofen", "Baclon", "Bacmax", "Beklo", "Bicozin",
     "Canazole", "Candinil", "Cetisoft", "Conaz", "Dancel", "Denixil", "Diflu",
     "Dinafex", "Disopan", "Esonix", "Esoral", "Etizin", "Exium", "Fenadin",
     "Fexofast", "Fexo", "Filmet", "Fixal", "Flamyd", "Flexibac", "Flexilax",