JJS341 commited on
Commit
f00324b
·
verified ·
1 Parent(s): c50e40a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -1,28 +1,29 @@
1
  import os
2
  import sys
3
 
4
- # 強制下載 Spacy 英文模型 (Lingmess 運作基礎)
5
  os.system(f"{sys.executable} -m spacy download en_core_web_sm")
6
 
7
- # 防錯補丁:萬一 huggingface_hub 版本太新,手動向舊路徑
8
- import huggingface_hub
9
- if not hasattr(huggingface_hub, 'HfFolder'):
10
- # 這是為了讓舊版 Gradio 能在新的 hub 環境下強行啟動
11
- class MockHfFolder:
12
- @staticmethod
13
- def get_token(): return os.getenv("HF_TOKEN")
14
- huggingface_hub.HfFolder = MockHfFolder
 
 
15
 
16
  import gradio as gr
17
  from fastcoref import LingmessCoref
18
  from deep_translator import GoogleTranslator
19
 
20
- # 初始化 Lingmess 模型
21
  model = LingmessCoref(device='cpu')
22
 
23
 
24
 
25
-
26
  def coref_chat(user_input):
27
  if not user_input.strip():
28
  return "請輸入內容", "等待輸入..."
 
1
  import os
2
  import sys
3
 
4
+ # 1. 強制下載 Spacy 模型
5
  os.system(f"{sys.executable} -m spacy download en_core_web_sm")
6
 
7
+ # 2. 防錯補丁:解決 huggingface_hub 版本太新導致的 ImportError
8
+ try:
9
+ import huggingface_hub
10
+ if not hasattr(huggingface_hub, 'HfFolder'):
11
+ class MockHfFolder:
12
+ @staticmethod
13
+ def get_token(): return os.getenv("HF_TOKEN")
14
+ huggingface_hub.HfFolder = MockHfFolder
15
+ except:
16
+ pass
17
 
18
  import gradio as gr
19
  from fastcoref import LingmessCoref
20
  from deep_translator import GoogleTranslator
21
 
22
+ # 3. 初始化 Lingmess 模型 (這行會跑比較久,因為要下載 Lingmess 權重)
23
  model = LingmessCoref(device='cpu')
24
 
25
 
26
 
 
27
  def coref_chat(user_input):
28
  if not user_input.strip():
29
  return "請輸入內容", "等待輸入..."