mohsen22 commited on
Commit
1f5c9b9
·
verified ·
1 Parent(s): a2f566b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +55 -0
app.py ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import logging
2
+ from telegram import Update
3
+ from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters
4
+ from transformers import AutoTokenizer, AutoModelForCausalLM
5
+ import torch
6
+
7
+ # تنظیمات لاگ
8
+ logging.basicConfig(
9
+ format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
10
+ level=logging.INFO
11
+ )
12
+
13
+ # بارگیری مدل و توکنایزر
14
+ model_name = "نام مدل شما" # نام مدل خود را اینجا وارد کنید
15
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
16
+ model = AutoModelForCausalLM.from_pretrained(model_name)
17
+
18
+ # تابع برای تولید پاسخ
19
+ def generate_response(input_text):
20
+ inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
21
+ output = model.generate(**inputs, max_new_tokens=100)
22
+ response = tokenizer.decode(output[0], skip_special_tokens=True)
23
+ return response
24
+
25
+ # تابع شروع
26
+ async def start(update: Update, context):
27
+ await update.message.reply_text('سلام! من یک بات هوش مصنوعی هستم. می‌توانید با من چت کنید.')
28
+
29
+ # تابع پاسخ به پیام‌ها
30
+ async def echo(update: Update, context):
31
+ user_message = update.message.text
32
+ response = generate_response(user_message)
33
+ await update.message.reply_text(response)
34
+
35
+ # تابع اصلی
36
+ def main():
37
+ # توکن بات تلگرام
38
+ token = "توکن بات شما" # توکن بات خود را اینجا وارد کنید
39
+
40
+ # ایجاد برنامه بات
41
+ application = ApplicationBuilder().token(token).build()
42
+
43
+ # اضافه کردن دستورات
44
+ start_handler = CommandHandler('start', start)
45
+ application.add_handler(start_handler)
46
+
47
+ # اضافه کردن handler برای پیام‌های معمولی
48
+ echo_handler = MessageHandler(filters.TEXT & ~filters.COMMAND, echo)
49
+ application.add_handler(echo_handler)
50
+
51
+ # شروع بات
52
+ application.run_polling()
53
+
54
+ if __name__ == '__main__':
55
+ main()