Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import easyocr
|
|
2 |
from gradio_client import Client, handle_file
|
3 |
import pandas as pd
|
4 |
import gradio as gr
|
5 |
-
import io
|
6 |
|
7 |
clientImg = Client("dj-dawgs-ipd/IPD-Image-ViT-Finetune")
|
8 |
clientEngText = Client("dj-dawgs-ipd/IPD-Text-English-Finetune")
|
@@ -18,12 +17,8 @@ def extract_text(image):
|
|
18 |
|
19 |
def predict(image):
|
20 |
|
21 |
-
img_byte_arr = io.BytesIO()
|
22 |
-
image.save(img_byte_arr, format='PNG')
|
23 |
-
img_byte_arr.seek(0)
|
24 |
-
|
25 |
imgResult = clientImg.predict(
|
26 |
-
|
27 |
api_name="/predict"
|
28 |
)
|
29 |
|
|
|
2 |
from gradio_client import Client, handle_file
|
3 |
import pandas as pd
|
4 |
import gradio as gr
|
|
|
5 |
|
6 |
clientImg = Client("dj-dawgs-ipd/IPD-Image-ViT-Finetune")
|
7 |
clientEngText = Client("dj-dawgs-ipd/IPD-Text-English-Finetune")
|
|
|
17 |
|
18 |
def predict(image):
|
19 |
|
|
|
|
|
|
|
|
|
20 |
imgResult = clientImg.predict(
|
21 |
+
image,
|
22 |
api_name="/predict"
|
23 |
)
|
24 |
|