diff --git a/Dockerfile b/Dockerfile
index 671699b..5b47bb2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,5 @@
-FROM python:3.8.13
+#FROM python:3.8.13
+FROM artifacts.iflytek.com/docker-private/atp/base_image_for_ailab:0.0.1
 
 WORKDIR /app
 
diff --git a/app.py b/app.py
index 8e7a864..f4a7664 100644
--- a/app.py
+++ b/app.py
@@ -1,3 +1,39 @@
+from transformers import pipeline
 import gradio as gr
+import cv2
+from PIL import Image
+from gradio.themes.utils import sizes
+theme = gr.themes.Default(radius_size=sizes.radius_none).set(
+    block_label_text_color = '#4D63FF',
+    block_title_text_color = '#4D63FF',
+    button_primary_text_color = '#4D63FF',
+    button_primary_background_fill='#FFFFFF',
+    button_primary_border_color='#4D63FF',
+    button_primary_background_fill_hover='#EDEFFF',
+)
 
-gr.Interface.load("models/nlpconnect/vit-gpt2-image-captioning").launch()
+
+image_to_text = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
+
+
+def ocr(image):
+    inp = Image.fromarray(image.astype('uint8'), 'RGB')
+    text = image_to_text(inp)
+
+    total_caption = "" 
+    for caption in text:
+        total_caption = total_caption + caption.get('generated_text')
+        total_caption = total_caption + '\r\n'
+
+    return total_caption 
+
+demo = gr.Interface(fn=ocr,
+                    inputs='image',
+                    outputs='text',
+                    title = "image2text",
+                    theme = theme,
+                    examples = ['soccer.jpg'])
+
+
+if __name__ == "__main__":
+    demo.queue(concurrency_count=10).launch()
diff --git a/requirements.txt b/requirements.txt
index 25acedd..7e2fba5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1 @@
-gradio
+Pillow
diff --git a/soccer.jpg b/soccer.jpg
new file mode 100644
index 0000000..3f2ec82
Binary files /dev/null and b/soccer.jpg differ