Sachi Wagaarachchi commited on
Commit
8fa49bb
·
1 Parent(s): 8fc941e
Files changed (4) hide show
  1. README.md +1 -1
  2. pyproject.toml +2 -2
  3. requirements.txt +2 -2
  4. 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.13.0
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
- "gradio",
13
  "transformers",
14
- "torch==2.2",
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
- gradio
2
  transformers
3
- torch==2.2
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):