dev-yuje commited on
Commit
b1040cf
ยท
1 Parent(s): 737169d

Hotfix: Gemini 404 and enhanced error visibility

Browse files
Files changed (1) hide show
  1. app.py +16 -3
app.py CHANGED
@@ -54,10 +54,16 @@ def load_all_models():
54
  try:
55
  # ๋ชจ๋ธ ๋‹ค์šด๋กœ๋“œ ๋ฐ ๋กœ๋“œ
56
  cached_model = hf_hub_download(repo_id=REPO_ID, filename=self.model_path, repo_type="space")
 
 
 
 
 
 
57
  self.model = tf.keras.models.load_model(cached_model, compile=False)
58
  self.load_error = "์„ฑ๊ณต"
59
  except Exception as model_e:
60
- self.load_error = f"๋ชจ๋ธ ๋‹ค์šด๋กœ๋“œ/๋กœ๋“œ ์‹คํŒจ: {model_e}\n{traceback.format_exc()}"
61
  except Exception as e:
62
  self.load_error = f"HuggingFace ๋ฆฌ์†Œ์Šค ๋กœ๋“œ ํ†ตํ•ฉ ์—๋Ÿฌ: {e}"
63
 
@@ -92,7 +98,13 @@ def load_all_models():
92
  class Consultant:
93
  def __init__(self):
94
  api_key = os.getenv("GOOGLE_API_KEY", "")
95
- self.llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", google_api_key=api_key, temperature=0.7)
 
 
 
 
 
 
96
  self.embedding_model = None
97
  self.retriever = None
98
 
@@ -147,7 +159,8 @@ def handle_predict(*args):
147
 
148
  res = _models["predictor"].predict(features_dict)
149
  if isinstance(res, str) and "Error" in res:
150
- return f"โŒ ๋ถ„์„ ์‹คํŒจ: {res}", f"โš ๏ธ ๋ถ„์„ ์—๋Ÿฌ ๋ฐœ์ƒ"
 
151
 
152
  score_val = str(int(round(float(res))))
153
  return {"features": features_dict, "score": score_val}, score_val
 
54
  try:
55
  # ๋ชจ๋ธ ๋‹ค์šด๋กœ๋“œ ๋ฐ ๋กœ๋“œ
56
  cached_model = hf_hub_download(repo_id=REPO_ID, filename=self.model_path, repo_type="space")
57
+ # [์ถ”๊ฐ€] ํŒŒ์ผ ํฌ๊ธฐ ํ™•์ธ ๋กœ๊ทธ (๋””๋ฒ„๊น…์šฉ)
58
+ fsize = os.path.getsize(cached_model)
59
+ if fsize < 1000:
60
+ self.load_error = f"ํŒŒ์ผ์ด ๋„ˆ๋ฌด ์ž‘์Œ({fsize}B). LFS ํฌ์ธํ„ฐ์ผ ๊ฐ€๋Šฅ์„ฑ ์žˆ์Œ."
61
+ return
62
+
63
  self.model = tf.keras.models.load_model(cached_model, compile=False)
64
  self.load_error = "์„ฑ๊ณต"
65
  except Exception as model_e:
66
+ self.load_error = f"๋ชจ๋ธ ๋กœ๋“œ ์‹คํŒจ: {str(model_e)}\n{traceback.format_exc()}"
67
  except Exception as e:
68
  self.load_error = f"HuggingFace ๋ฆฌ์†Œ์Šค ๋กœ๋“œ ํ†ตํ•ฉ ์—๋Ÿฌ: {e}"
69
 
 
98
  class Consultant:
99
  def __init__(self):
100
  api_key = os.getenv("GOOGLE_API_KEY", "")
101
+ # [์ˆ˜์ •] ๋ชจ๋ธ ๋ช…์นญ ๋ฐ API ๋ฒ„์ „ ํ˜ธํ™˜์„ฑ ๊ณ ๋ ค
102
+ self.llm = ChatGoogleGenerativeAI(
103
+ model="gemini-1.5-flash",
104
+ google_api_key=api_key,
105
+ temperature=0.7,
106
+ convert_system_message_to_human=True
107
+ )
108
  self.embedding_model = None
109
  self.retriever = None
110
 
 
159
 
160
  res = _models["predictor"].predict(features_dict)
161
  if isinstance(res, str) and "Error" in res:
162
+ # ์—๋Ÿฌ ๋ฐœ์ƒ ์‹œ UI ์ƒ๋‹จ ์ ์ˆ˜์นธ์— ์—๋Ÿฌ ๋‚ด์šฉ ๋Œ€๋Ÿ‰ ๋…ธ์ถœ (์ง„๋ฒ” ํ™•์ธ์šฉ)
163
+ return f"โŒ ๋ถ„์„ ์‹คํŒจ: {res}", f"โš ๏ธ {res[:100]}..."
164
 
165
  score_val = str(int(round(float(res))))
166
  return {"features": features_dict, "score": score_val}, score_val