Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ output: Potential widening of the achievement gap if data is not used equitably.
|
|
23 |
def generate(text):
|
24 |
try:
|
25 |
prompt = system_prompt + f"\ninput: {text}\noutput:"
|
26 |
-
inputs = tokenizer.encode(prompt, return_tensors="pt", return_attention_mask=True
|
27 |
outputs = model.generate(inputs["input_ids"], attention_mask=inputs["attention_mask"], max_length=256)
|
28 |
response_text = tokenizer.decode(outputs[0], skip_special_tokens=True).split("output:")[-1].strip()
|
29 |
return response_text if response_text else "No valid response generated."
|
|
|
23 |
def generate(text):
|
24 |
try:
|
25 |
prompt = system_prompt + f"\ninput: {text}\noutput:"
|
26 |
+
inputs = tokenizer.encode(prompt, return_tensors="pt", return_attention_mask=True)
|
27 |
outputs = model.generate(inputs["input_ids"], attention_mask=inputs["attention_mask"], max_length=256)
|
28 |
response_text = tokenizer.decode(outputs[0], skip_special_tokens=True).split("output:")[-1].strip()
|
29 |
return response_text if response_text else "No valid response generated."
|