themissingCRAM
commited on
Commit
·
e5190f1
1
Parent(s):
c2cfe36
transformerModel
Browse files- README.md +3 -2
- app.py +2 -2
- requirements.txt +1 -1
README.md
CHANGED
|
@@ -1,13 +1,14 @@
|
|
| 1 |
---
|
| 2 |
-
title: bakery_shop_ordering_system_with_recipe_rag
|
| 3 |
emoji: 🎷
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 5.23.3
|
| 8 |
app_file: app.py
|
| 9 |
-
pinned:
|
| 10 |
short_description: self correcting text to sql agent based on smolagents exampl
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
bakery shops ordering system with recipe rag
|
|
|
|
| 1 |
---
|
| 2 |
+
title: bakery_shop_ordering_system_with_recipe_rag
|
| 3 |
emoji: 🎷
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 5.23.3
|
| 8 |
app_file: app.py
|
| 9 |
+
pinned: True
|
| 10 |
short_description: self correcting text to sql agent based on smolagents exampl
|
| 11 |
+
|
| 12 |
---
|
| 13 |
|
| 14 |
bakery shops ordering system with recipe rag
|
app.py
CHANGED
|
@@ -160,8 +160,8 @@ if __name__ == "__main__":
|
|
| 160 |
|
| 161 |
# you can use HFApiModel from smolagents if you don't mind paying more money.
|
| 162 |
model = TransformersModel(
|
| 163 |
-
model_id="Qwen/Qwen2.5-Coder-14B-Instruct",
|
| 164 |
-
|
| 165 |
device_map="cuda"
|
| 166 |
,max_new_tokens=5000,torch_dtype=torch.bfloat16
|
| 167 |
)
|
|
|
|
| 160 |
|
| 161 |
# you can use HFApiModel from smolagents if you don't mind paying more money.
|
| 162 |
model = TransformersModel(
|
| 163 |
+
# model_id="Qwen/Qwen2.5-Coder-14B-Instruct",
|
| 164 |
+
model_id="meta-llama/Llama-3.2-3B-Instruct",
|
| 165 |
device_map="cuda"
|
| 166 |
,max_new_tokens=5000,torch_dtype=torch.bfloat16
|
| 167 |
)
|
requirements.txt
CHANGED
|
@@ -11,4 +11,4 @@ langchain_community == 0.3.20
|
|
| 11 |
chromadb == 0.6.3
|
| 12 |
transformers == 4.50.3
|
| 13 |
accelerate >=0.26.0
|
| 14 |
-
torch
|
|
|
|
| 11 |
chromadb == 0.6.3
|
| 12 |
transformers == 4.50.3
|
| 13 |
accelerate >=0.26.0
|
| 14 |
+
torch
|