Spaces:
Running
Running
Aleksey Khlopkov
commited on
Commit
·
99bafa0
1
Parent(s):
74d4655
replying to msg
Browse files- .gitignore +1 -1
- main.py +5 -4
- requirements.txt +22 -0
.gitignore
CHANGED
@@ -173,7 +173,7 @@ cython_debug/
|
|
173 |
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
174 |
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
175 |
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
176 |
-
|
177 |
|
178 |
# Abstra
|
179 |
# Abstra is an AI-powered process automation framework.
|
|
|
173 |
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
174 |
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
175 |
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
176 |
+
.idea/
|
177 |
|
178 |
# Abstra
|
179 |
# Abstra is an AI-powered process automation framework.
|
main.py
CHANGED
@@ -8,7 +8,6 @@ from dotenv import load_dotenv
|
|
8 |
from models.seq2seq.model import Seq2SeqChatbot
|
9 |
import torch
|
10 |
|
11 |
-
|
12 |
load_dotenv()
|
13 |
|
14 |
TOKEN: Final = os.environ.get("TOKEN")
|
@@ -20,10 +19,11 @@ CHECKPOINT_PATH: Final = "models/seq2seq/checkpoint/150_checkpoint.tar"
|
|
20 |
torch.manual_seed(0)
|
21 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
22 |
|
23 |
-
chatbot = Seq2SeqChatbot(500, 2, 2, 0.1, device)
|
24 |
chatbot.load_checkpoint(CHECKPOINT_PATH)
|
25 |
chatbot.eval_mode()
|
26 |
|
|
|
27 |
def handle_response(text: str) -> Optional[str]:
|
28 |
response_chance = 1.0
|
29 |
if random.random() < response_chance:
|
@@ -36,7 +36,7 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
36 |
text: str = update.message.text.replace(BOT_USERNAME, '').strip().lower()
|
37 |
response: Optional[str] = handle_response(text)
|
38 |
if response:
|
39 |
-
await context.bot.sendMessage(update.message.chat_id, response)
|
40 |
|
41 |
|
42 |
async def error(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
@@ -44,6 +44,7 @@ async def error(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
44 |
f"chat caused error \"{context.error}\"\n"
|
45 |
f"{update}\"")
|
46 |
|
|
|
47 |
def main() -> None:
|
48 |
"""Run the bot."""
|
49 |
application = Application.builder().token(TOKEN).build()
|
@@ -57,4 +58,4 @@ def main() -> None:
|
|
57 |
if __name__ == '__main__':
|
58 |
print("Running main...")
|
59 |
# print(chatbot("test"))
|
60 |
-
main()
|
|
|
8 |
from models.seq2seq.model import Seq2SeqChatbot
|
9 |
import torch
|
10 |
|
|
|
11 |
load_dotenv()
|
12 |
|
13 |
TOKEN: Final = os.environ.get("TOKEN")
|
|
|
19 |
torch.manual_seed(0)
|
20 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
21 |
|
22 |
+
chatbot = Seq2SeqChatbot(500, 8856, 2, 2, 0.1, device)
|
23 |
chatbot.load_checkpoint(CHECKPOINT_PATH)
|
24 |
chatbot.eval_mode()
|
25 |
|
26 |
+
|
27 |
def handle_response(text: str) -> Optional[str]:
|
28 |
response_chance = 1.0
|
29 |
if random.random() < response_chance:
|
|
|
36 |
text: str = update.message.text.replace(BOT_USERNAME, '').strip().lower()
|
37 |
response: Optional[str] = handle_response(text)
|
38 |
if response:
|
39 |
+
await context.bot.sendMessage(update.message.chat_id, response, reply_to_message_id=update.message.id)
|
40 |
|
41 |
|
42 |
async def error(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
|
44 |
f"chat caused error \"{context.error}\"\n"
|
45 |
f"{update}\"")
|
46 |
|
47 |
+
|
48 |
def main() -> None:
|
49 |
"""Run the bot."""
|
50 |
application = Application.builder().token(TOKEN).build()
|
|
|
58 |
if __name__ == '__main__':
|
59 |
print("Running main...")
|
60 |
# print(chatbot("test"))
|
61 |
+
main()
|
requirements.txt
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
anyio==4.9.0
|
2 |
+
certifi==2025.7.14
|
3 |
+
colorama==0.4.6
|
4 |
+
filelock==3.18.0
|
5 |
+
fsspec==2025.7.0
|
6 |
+
h11==0.16.0
|
7 |
+
httpcore==1.0.9
|
8 |
+
httpx==0.28.1
|
9 |
+
idna==3.10
|
10 |
+
Jinja2==3.1.6
|
11 |
+
MarkupSafe==3.0.2
|
12 |
+
mpmath==1.3.0
|
13 |
+
networkx==3.5
|
14 |
+
numpy==2.3.1
|
15 |
+
python-dotenv==1.1.1
|
16 |
+
python-telegram-bot==22.3
|
17 |
+
setuptools==80.9.0
|
18 |
+
sniffio==1.3.1
|
19 |
+
sympy==1.14.0
|
20 |
+
torch==2.7.1
|
21 |
+
tqdm==4.67.1
|
22 |
+
typing_extensions==4.14.1
|