seawolf2357 commited on
Commit
a0d56d8
·
verified ·
1 Parent(s): 265c54d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -22,8 +22,10 @@ intents.messages = True
22
  intents.guilds = True
23
  intents.guild_messages = True
24
 
25
- # Cohere 모델 설정
26
- hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
 
 
27
 
28
  # PaliGemma 모델 설정 (CPU 모드)
29
  model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner").to("cpu").eval()
 
22
  intents.guilds = True
23
  intents.guild_messages = True
24
 
25
+
26
+ # 추론 API 클라이언트 설정
27
+ # hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
28
+ hf_client = InferenceClient("CohereForAI/aya-23-35B", token=os.getenv("HF_TOKEN"))
29
 
30
  # PaliGemma 모델 설정 (CPU 모드)
31
  model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner").to("cpu").eval()