Rid3 commited on
Commit
15940e2
·
verified ·
1 Parent(s): e6e1be1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -10
app.py CHANGED
@@ -1,23 +1,40 @@
 
1
  import telebot
2
- from g4f.client import Client
 
 
 
 
 
 
3
 
4
  # Твой токен от @BotFather
5
- bot = telebot.TeleBot("8531345451:AAE8qJlFEwIcuQdUiIFk2viZS9xLLMk2c_o")
6
- client = Client()
 
 
 
7
 
8
  @bot.message_handler(content_types=['text'])
9
  def handle_message(message):
10
  try:
11
- # Запрашиваем ответ у бесплатного провайдера одель можно менять, например, на gpt-4o или llama-3)
12
- response = client.chat.completions.create(
13
- model="gpt-4o",
14
- messages=[{"role": "user", "content": message.text}]
 
 
 
 
15
  )
16
- bot.reply_to(message, response.choices[0].message.content)
 
 
 
17
  except Exception as e:
18
- bot.reply_to(message, f"Ошибка: {e}")
19
 
20
  if __name__ == "__main__":
21
- print("Бот запущен...")
22
  bot.infinity_polling()
23
 
 
1
+ import os
2
  import telebot
3
+ from huggingface_hub import InferenceClient
4
+
5
+ # Внутри Spaces токен HF уже зашит в переменные окружения, вытягиваем его автоматически
6
+ HF_TOKEN = os.environ.get("HF_TOKEN")
7
+
8
+ # Если вдруг запускаешь локально и токена нет, клиент попытается работать без него (на публичных лимитах)
9
+ client = InferenceClient(token=HF_TOKEN)
10
 
11
  # Твой токен от @BotFather
12
+ TG_TOKEN = "8531345451:AAE8qJlFEwIcuQdUiIFk2viZS9xLLMk2c_o"
13
+ bot = telebot.TeleBot(TG_TOKEN)
14
+
15
+ # Выбираем стабильную бесплатную модель от Meta
16
+ MODEL_NAME = "meta-llama/Meta-Llama-3-8B-Instruct"
17
 
18
  @bot.message_handler(content_types=['text'])
19
  def handle_message(message):
20
  try:
21
+ # Отправляем запрос в официальный бесплатный API Hugging Face
22
+ response = client.chat_completion(
23
+ model=MODEL_NAME,
24
+ messages=[
25
+ {"role": "system", "content": "Ты умный AI ассистент. Отвечай четко и только на русском языке."},
26
+ {"role": "user", "content": message.text}
27
+ ],
28
+ max_tokens=512
29
  )
30
+
31
+ bot_reply = response.choices[0].message.content
32
+ bot.reply_to(message, bot_reply)
33
+
34
  except Exception as e:
35
+ bot.reply_to(message, f"Ошибка нейросети: {str(e)}")
36
 
37
  if __name__ == "__main__":
38
+ print("Бот успешно запущен на мощностях HF!")
39
  bot.infinity_polling()
40