EMMA commited on
Commit
ff9480b
·
1 Parent(s): 0d6817f

Fix gradio version conflict

Browse files
Files changed (1) hide show
  1. app.py +16 -6
app.py CHANGED
@@ -55,17 +55,27 @@ tokenizer = tiktoken.get_encoding("cl100k_base")
55
  GAPGPT_BASE_URL = os.getenv("GAPGPT_BASE_URL", "https://api.gapgpt.app/v1")
56
  openai_client_cache = {}
57
 
58
- LOCAL_MODEL_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Classification_Model", "final_xlm_r_model_router")
 
59
 
60
- print(f"Loading model from: {LOCAL_MODEL_PATH}")
61
 
62
  try:
63
- _tokenizer = AutoTokenizer.from_pretrained(LOCAL_MODEL_PATH)
64
- _classifier_model = AutoModelForSequenceClassification.from_pretrained(LOCAL_MODEL_PATH)
 
 
 
 
 
 
65
  _classifier_model.eval()
66
- print("Local classifier loaded successfully.")
67
  except Exception as e:
68
- print(f"Error loading local classifier: {e}")
 
 
 
69
 
70
 
71
  def get_gapgpt_client(api_key: str) -> OpenAIClient:
 
55
  GAPGPT_BASE_URL = os.getenv("GAPGPT_BASE_URL", "https://api.gapgpt.app/v1")
56
  openai_client_cache = {}
57
 
58
+ HF_MODEL_ID = "Keyvan1986/Emma-Classification_Model"
59
+ HF_TOKEN = os.getenv("HF_TOKEN") # در صورت Private بودن مدل
60
 
61
+ print(f"Loading classifier model from Hugging Face: {HF_MODEL_ID}")
62
 
63
  try:
64
+ _tokenizer = AutoTokenizer.from_pretrained(
65
+ HF_MODEL_ID,
66
+ token=HF_TOKEN
67
+ )
68
+ _classifier_model = AutoModelForSequenceClassification.from_pretrained(
69
+ HF_MODEL_ID,
70
+ token=HF_TOKEN
71
+ )
72
  _classifier_model.eval()
73
+ print("Classifier model loaded successfully from Hugging Face Hub.")
74
  except Exception as e:
75
+ print(f"Error loading classifier from Hugging Face: {e}")
76
+ _tokenizer = None
77
+ _classifier_model = None
78
+
79
 
80
 
81
  def get_gapgpt_client(api_key: str) -> OpenAIClient: