Update prediction.py

prediction.py

@@ -14,9 +14,9 @@ import pickle
 
 #load model
 def run():
-    file = st.file_uploader("Upload an image", type=["jpg", "png"])
+    file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
 
-    model = load_model('model_transfer.h5', custom_objects={'KerasLayer': hub.KerasLayer})
+    model = load_model('model_tl.h5', custom_objects={'KerasLayer': hub.KerasLayer})
     target_size=(224, 224)
 
     def import_and_predict(image_data, model):
@@ -32,9 +32,9 @@ def run():
         classes = model.predict(x)
         result_pred = np.where(classes < 0.7, 0, 1)
         if result_pred == 1:
-            result = 'Brain Tumor'
+            result = 'Pneumonia'
         else:
-            result = 'Healthy'
+            result = 'Normal'
 
         return f"Prediction: {result}"