siddhantuniyal commited on
Commit
5f6d298
·
verified ·
1 Parent(s): 0edcf56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
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
- img_byte_arr,
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