bwbai commited on
Commit
cffe169
·
1 Parent(s): f3ca234

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -2,6 +2,7 @@
2
  import os
3
  import gradio as gr
4
  import openai
 
5
 
6
  #from dotenv import load_dotenv
7
 
@@ -34,6 +35,8 @@ def custom_chat(user_input):
34
  return chat_reply
35
 
36
  def dothis(file):
 
 
37
  return str(file.name)
38
 
39
  #demo = gr.Interface(fn=custom_chat,
 
2
  import os
3
  import gradio as gr
4
  import openai
5
+ import ocrmypdf
6
 
7
  #from dotenv import load_dotenv
8
 
 
35
  return chat_reply
36
 
37
  def dothis(file):
38
+ if __name__ == '__main__': # To ensure correct behavior on Windows and macOS
39
+ out = ocrmypdf.ocr( file, 'output.pdf', deskew=True)
40
  return str(file.name)
41
 
42
  #demo = gr.Interface(fn=custom_chat,