Update app.py

app.py

@@ -44,7 +44,7 @@ def pre_process(image, model_name):
     img = image.convert("L")
     
     # Select model based on the dropdown selection
-    if model_name == "cnn":
+    if model_name == "Simple CNN":
         model = mnist_cnn_model
         img = img.resize((28, 28))
         img = np.array(img) / 255.0
@@ -52,7 +52,7 @@ def pre_process(image, model_name):
         pred = model.predict(final_img)
         result = np.argmax(pred)
         return str(result)
-    elif model_name == "mcdnn":
+    elif model_name == "MCDNN":
         model = mnist_mcdnn_model
         img = img.resize((28, 28))
         img = np.array(img) / 255.0
@@ -60,16 +60,16 @@ def pre_process(image, model_name):
         pred = model.predict(final_img)
         result = np.argmax(pred)
         return str(result)
-    elif model_name == "OCR - preferred":
+    elif model_name == "OCR":
         pred = run(image)
         return pred
-    elif model_name == "TROCR":
+    elif model_name == "CNN BI-LSTM":
         return trocr_predict(image)
     else:
         return "Invalid model selection"
 
 # Defining model options for dropdown
-model_options = ["cnn", "mcdnn", "OCR - preferred", "TROCR"]
+model_options = ["Simple CNN", "MCDNN", "OCR", "CNN BI-LSTM"]
 
 
 with gr.Blocks() as demo: