Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -56,13 +56,12 @@ Documents: {documents}
|
|
| 56 |
Question: {user_message}
|
| 57 |
Answer:
|
| 58 |
""",
|
| 59 |
-
input_variables=["user_message", "documents"]
|
| 60 |
)
|
| 61 |
|
| 62 |
llm = HuggingFacePipeline.from_model_id(
|
| 63 |
model_id="meta-llama/Llama-3.2-1B-Instruct",
|
| 64 |
task="text-generation",
|
| 65 |
-
device = 0,
|
| 66 |
pipeline_kwargs=dict(
|
| 67 |
repetition_penalty=1.5,
|
| 68 |
temperature=0.3,
|
|
|
|
| 56 |
Question: {user_message}
|
| 57 |
Answer:
|
| 58 |
""",
|
| 59 |
+
input_variables=["user_message", "documents"]
|
| 60 |
)
|
| 61 |
|
| 62 |
llm = HuggingFacePipeline.from_model_id(
|
| 63 |
model_id="meta-llama/Llama-3.2-1B-Instruct",
|
| 64 |
task="text-generation",
|
|
|
|
| 65 |
pipeline_kwargs=dict(
|
| 66 |
repetition_penalty=1.5,
|
| 67 |
temperature=0.3,
|