xiaoyao9184 commited on
Commit
5e5f874
·
verified ·
1 Parent(s): 85077e2

Synced repo using 'sync_with_huggingface' Github Action

Browse files
Files changed (1) hide show
  1. gradio_app.py +12 -7
gradio_app.py CHANGED
@@ -2,9 +2,12 @@ import os
2
  import sys
3
 
4
  if "APP_PATH" in os.environ:
5
- os.chdir(os.environ["APP_PATH"])
6
- # fix sys.path for import
7
- sys.path.append(os.getcwd())
 
 
 
8
 
9
  os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1" # For some reason, transformers decided to use .isin for a simple op, which is not supported on MPS
10
 
@@ -136,9 +139,10 @@ def del_select_coordinates(img, evt: gr.SelectData):
136
 
137
  return (img[0], sections)
138
 
139
-
140
- model = load_model_cached()
141
- processor = load_processor_cached()
 
142
 
143
  with gr.Blocks(title="Texify") as demo:
144
  gr.Markdown("""
@@ -215,4 +219,5 @@ with gr.Blocks(title="Texify") as demo:
215
  outputs=[markdown_result]
216
  )
217
 
218
- demo.launch()
 
 
2
  import sys
3
 
4
  if "APP_PATH" in os.environ:
5
+ app_path = os.path.abspath(os.environ["APP_PATH"])
6
+ if os.getcwd() != app_path:
7
+ # fix sys.path for import
8
+ os.chdir(app_path)
9
+ if app_path not in sys.path:
10
+ sys.path.append(app_path)
11
 
12
  os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1" # For some reason, transformers decided to use .isin for a simple op, which is not supported on MPS
13
 
 
139
 
140
  return (img[0], sections)
141
 
142
+ # Load models if not already loaded in reload mode
143
+ if 'model' not in globals():
144
+ model = load_model_cached()
145
+ processor = load_processor_cached()
146
 
147
  with gr.Blocks(title="Texify") as demo:
148
  gr.Markdown("""
 
219
  outputs=[markdown_result]
220
  )
221
 
222
+ if __name__ == "__main__":
223
+ demo.launch()