Upload 2 files

ocr_gpt_app.py

@@ -0,0 +1,24 @@
+import easyocr
+import gradio as gr
+from transformers import pipeline
+
+# Initialize OCR reader and GPT pipeline
+reader = easyocr.Reader(['ch_sim', 'en'])
+gpt = pipeline('text-generation', model='gpt2')
+
+# Define the OCR function
+def ocr_gpt(image):
+    ocr_result = reader.readtext(image)
+    corrected_text = gpt(ocr_result[0][1])[0]['generated_text']
+    gpt_opinion = gpt("What do you think about the information in the image?")[0]['generated_text']
+    return corrected_text, gpt_opinion
+
+# Define Gradio UI components
+image_input = gr.inputs.Image()
+gpt_input = gr.inputs.Textbox(lines=1, label="GPT API Input")
+ocr_output = gr.outputs.Textbox(label="OCR Result (GPT Corrected)")
+gpt_opinion_output = gr.outputs.Textbox(label="GPT Opinion on Image Information")
+
+# Create Gradio interface
+iface = gr.Interface(fn=ocr_gpt, inputs=[gpt_input, image_input], outputs=[ocr_output, gpt_opinion_output])
+#iface.launch(share=True)

requirements.txt

@@ -0,0 +1,3 @@
+easyocr
+gradio
+transformers