Sachi Wagaarachchi
commited on
Commit
·
8fa49bb
1
Parent(s):
8fc941e
updated reqs, [email protected]
Browse files- README.md +1 -1
- pyproject.toml +2 -2
- requirements.txt +2 -2
- src/chat_logic.py +2 -0
README.md
CHANGED
@@ -4,7 +4,7 @@ emoji: 💬
|
|
4 |
colorFrom: yellow
|
5 |
colorTo: purple
|
6 |
sdk: gradio
|
7 |
-
sdk_version: 5.
|
8 |
app_file: app.py
|
9 |
pinned: true
|
10 |
license: apache-2.0
|
|
|
4 |
colorFrom: yellow
|
5 |
colorTo: purple
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 5.29.0
|
8 |
app_file: app.py
|
9 |
pinned: true
|
10 |
license: apache-2.0
|
pyproject.toml
CHANGED
@@ -9,9 +9,9 @@ version = "0.1.0"
|
|
9 |
description = "Gradio app with Qwen models"
|
10 |
requires-python = ">=3.10"
|
11 |
dependencies = [
|
12 |
-
"
|
13 |
"transformers",
|
14 |
-
"torch
|
15 |
"accelerate",
|
16 |
"huggingface_hub",
|
17 |
"gradio",
|
|
|
9 |
description = "Gradio app with Qwen models"
|
10 |
requires-python = ">=3.10"
|
11 |
dependencies = [
|
12 |
+
"numpy>=2.2",
|
13 |
"transformers",
|
14 |
+
"torch",
|
15 |
"accelerate",
|
16 |
"huggingface_hub",
|
17 |
"gradio",
|
requirements.txt
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
-
|
2 |
transformers
|
3 |
-
torch
|
4 |
accelerate
|
5 |
huggingface_hub
|
6 |
gradio
|
|
|
1 |
+
numpy>=2.2
|
2 |
transformers
|
3 |
+
torch
|
4 |
accelerate
|
5 |
huggingface_hub
|
6 |
gradio
|
src/chat_logic.py
CHANGED
@@ -2,7 +2,9 @@ from transformers import TextIteratorStreamer
|
|
2 |
import threading
|
3 |
from src.utils import format_prompt
|
4 |
import logging
|
|
|
5 |
|
|
|
6 |
class ChatProcessor:
|
7 |
"""Processes chat interactions using Qwen models"""
|
8 |
def __init__(self, model_manager, vector_db):
|
|
|
2 |
import threading
|
3 |
from src.utils import format_prompt
|
4 |
import logging
|
5 |
+
from spaces import *
|
6 |
|
7 |
+
@spaces.GPU
|
8 |
class ChatProcessor:
|
9 |
"""Processes chat interactions using Qwen models"""
|
10 |
def __init__(self, model_manager, vector_db):
|