Update app.py

app.py

@@ -5,10 +5,7 @@ from huggingface_hub import InferenceClient
 HF_TOKEN = st.secrets["HF_TOKEN"]
 
 # Inference Client 설정 (GRIN-MoE 모델 사용)
-client = InferenceClient(
-    repo_id="microsoft/GRIN-MoE",
-    token=HF_TOKEN
-)
+client = InferenceClient(token=HF_TOKEN)
 
 # Streamlit 페이지 설정
 st.set_page_config(page_title="GRIN-MoE AI Chat", page_icon="🤖")
@@ -25,6 +22,7 @@ user_input = st.text_input("질문을 입력하세요:")
 def generate_streaming_response(prompt):
     response_text = ""
     for message in client.chat_completion(
+        model="microsoft/GRIN-MoE",  # 모델 이름을 명시적으로 전달
         messages=[{"role": "user", "content": prompt}],
         max_tokens=500,
         stream=True
@@ -54,3 +52,4 @@ if st.session_state.messages:
         else:
             st.write(f"**AI:** {msg['content']}")
 
+