Update app.py

Updated predict_text() to Resize Segments

app.py

@@ -20,15 +20,20 @@ def preprocess_image(image):
 # Function to predict text from handwritten prescription
 def predict_text(image):
     image = image.convert("RGB")
-    image = image.resize((128, 64))  # Resize for better character separation
+    image = image.resize((128, 64))  # Resize for better character segmentation
     image = np.array(image) / 255.0  # Normalize
 
-    num_chars = 5  # Estimate word length
+    num_chars = 5  # Estimated number of characters in the word
     segment_width = image.shape[1] // num_chars  # Split image into equal parts
 
     predicted_text = []
     for i in range(num_chars):
         char_segment = image[:, i * segment_width:(i + 1) * segment_width, :]
+
+        # Resize each character segment to (64, 64, 3) to match model input
+        char_segment = Image.fromarray((char_segment * 255).astype(np.uint8))  # Convert back to PIL
+        char_segment = char_segment.resize((64, 64))  # Resize to match model input
+        char_segment = np.array(char_segment) / 255.0  # Normalize again
         char_segment = np.expand_dims(char_segment, axis=0)  # Add batch dimension
 
         # Predict character for each segment