Update app.py

app.py

@@ -35,8 +35,11 @@ def predict():
     if not file:
         return jsonify({'error': 'No image provided'}), 400
 
-    img = preprocess_image(file.read())
-    pred = model.predict(img)[0]
+    img = preprocess_image(file.read()).astype(np.float32)
+    interpreter.set_tensor(input_details[0]['index'], img)
+    interpreter.invoke()
+    pred = interpreter.get_tensor(output_details[0]['index'])[0]
+
     sorted_indices = np.argsort(pred)[::-1]
     top1, top2 = sorted_indices[:2]
     top1_label = class_names[top1]