Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -145,7 +145,7 @@ def generate_llama_response(query, context=None):
|
|
| 145 |
input_text += f"Context: {context}\n"
|
| 146 |
input_text += "Answer:"
|
| 147 |
inputs = tokenizer(input_text, return_tensors="pt")
|
| 148 |
-
outputs = model.generate(inputs["input_ids"], max_length=
|
| 149 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 150 |
return response
|
| 151 |
|
|
|
|
| 145 |
input_text += f"Context: {context}\n"
|
| 146 |
input_text += "Answer:"
|
| 147 |
inputs = tokenizer(input_text, return_tensors="pt")
|
| 148 |
+
outputs = model.generate(inputs["input_ids"], max_length=40, temperature=0.7)
|
| 149 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 150 |
return response
|
| 151 |
|