Add application file

app.py

@@ -18,7 +18,7 @@ from happytransformer import HappyTextToText, TTSettings
 
 configs = BaseModelConfigs.load("./configs.yaml")
 
-happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
+#happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
 
 beam_settings = TTSettings(num_beams=5, min_length=1, max_length=100)
 
@@ -67,9 +67,9 @@ async def predict_text(image: UploadFile):
 
     # Make a prediction
     extracted_text = model.predict(img)
-    corrected_text = happy_tt.generate_text(extracted_text, beam_settings)
+    #corrected_text = happy_tt.generate_text(extracted_text, beam_settings)
 
-    return {"text": extracted_text, "corrected_text": corrected_text}
+    return {"text": extracted_text}
 
 
 @app.post("/extract_text/")
@@ -83,9 +83,9 @@ async def extract_text_from_image(image: UploadFile):
 
         # Perform OCR on the image
         extracted_text = pytesseract.image_to_string(img)
-        corrected_text = happy_tt.generate_text(extracted_text, beam_settings)
+        #corrected_text = happy_tt.generate_text(extracted_text, beam_settings)
 
-        return {"text": extracted_text, "corrected_text": corrected_text}
+        return {"text": extracted_text}
     else:
         return {"error": "Invalid file format. Please upload an image."}