Update app.py

app.py

@@ -31,9 +31,7 @@ from sklearn.feature_extraction.text import TfidfVectorizer
 
 
 
-processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
-model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
-plt.switch_backend('Agg')
+
 def horizontal_projections(sobel_image):
     return np.sum(sobel_image, axis=1)  
             
@@ -304,16 +302,17 @@ def extract(image):
         print(line_image)
         new_p = Image.fromarray(line_image)
         new_p = new_p.convert("L")
-        imageio.imwrite('demo.jpeg',new_p)
+        imageio.imwrite('demo1.jpeg',new_p)
 
 
 
-        image = Image.open("demo.jpeg").convert("RGB")
+        image = Image.open("demo1.jpeg").convert("RGB")
         
         #print("Started Processing")
         #image = line_image
         pixel_values = processor(images=image, return_tensors="pt").pixel_values
-
+        processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
+        model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
         generated_ids = model.generate(pixel_values)
         generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
         print(generated_text)