parent
c829a8c2fd
commit
bfb92fe337
@ -0,0 +1,41 @@
|
|||||||
|
from gradio_client import Client
|
||||||
|
#client = Client("https://skier8402-mistral-super-fast.hf.space/")
|
||||||
|
|
||||||
|
from deep_translator import GoogleTranslator
|
||||||
|
|
||||||
|
def predict(prompt, client):
|
||||||
|
global iddb
|
||||||
|
result = client.predict(
|
||||||
|
prompt,
|
||||||
|
0.5, # 'Temperature'
|
||||||
|
256, # 'Max new tokens'
|
||||||
|
0.9, # 'Top-p (nucleus sampling)'
|
||||||
|
1.2, # 'Repetition penalty'
|
||||||
|
api_name="/chat"
|
||||||
|
)
|
||||||
|
return result
|
||||||
|
|
||||||
|
# text IN language IN
|
||||||
|
def translate(text, source):
|
||||||
|
if source == "ru":
|
||||||
|
target = "en"
|
||||||
|
elif source == "en":
|
||||||
|
target = "ru"
|
||||||
|
|
||||||
|
out = GoogleTranslator(source = source, target = target).translate(text)
|
||||||
|
return out
|
||||||
|
|
||||||
|
|
||||||
|
iddb = {}
|
||||||
|
|
||||||
|
def gen(text, id):
|
||||||
|
global iddb
|
||||||
|
if str(id) not in iddb:
|
||||||
|
client = Client("https://skier8402-mistral-super-fast.hf.space/")
|
||||||
|
iddb[str(id)] = client
|
||||||
|
else:
|
||||||
|
client = iddb[str(id)]
|
||||||
|
|
||||||
|
prompt = translate(text, "ru")
|
||||||
|
predicted = translate( predict(prompt, client), "en" ).replace("</s>", "")
|
||||||
|
return predicted
|
@ -0,0 +1,38 @@
|
|||||||
|
import telebot
|
||||||
|
|
||||||
|
### LOAD TOKEN ###
|
||||||
|
import json, os
|
||||||
|
|
||||||
|
if not os.path.exists('db.json'):
|
||||||
|
db = {"token": "None"}
|
||||||
|
js = json.dumps(db, indent=2)
|
||||||
|
with open("db.json", "w") as outfile:
|
||||||
|
outfile.write(js)
|
||||||
|
|
||||||
|
print('Input token in "None" (db.json)')
|
||||||
|
exit()
|
||||||
|
with open('db.json', 'r') as openfile:
|
||||||
|
db = json.load(openfile)
|
||||||
|
|
||||||
|
API_TOKEN = db["token"]
|
||||||
|
bot = telebot.TeleBot(API_TOKEN)
|
||||||
|
##################
|
||||||
|
|
||||||
|
@bot.message_handler(commands=['help', 'start'])
|
||||||
|
def send_welcome(message):
|
||||||
|
bot.reply_to(message, "None")
|
||||||
|
|
||||||
|
|
||||||
|
### MAIN ###
|
||||||
|
from api import *
|
||||||
|
|
||||||
|
|
||||||
|
@bot.message_handler(func=lambda message: True)
|
||||||
|
def echo_message(message):
|
||||||
|
bot.send_chat_action(message.chat.id, "typing", 30)
|
||||||
|
bot.reply_to(message, gen(message.text, message.chat.id))
|
||||||
|
|
||||||
|
|
||||||
|
############
|
||||||
|
|
||||||
|
bot.infinity_polling()
|
Loading…
Reference in new issue