Spaces:
Runtime error
Runtime error
remove debug note
Browse files
app.py
CHANGED
|
@@ -43,7 +43,7 @@ def generate_response(prompt, max_length=1024):
|
|
| 43 |
|
| 44 |
outputs = pipe(messages, max_new_tokens=max_length)
|
| 45 |
|
| 46 |
-
print("Raw output:", outputs)
|
| 47 |
# Extract just the assistant's response
|
| 48 |
try:
|
| 49 |
# outputs[0]["generated_text"] is already a list of messages
|
|
@@ -53,7 +53,7 @@ def generate_response(prompt, max_length=1024):
|
|
| 53 |
if assistant_message["role"] == "assistant":
|
| 54 |
return assistant_message["content"]
|
| 55 |
except Exception as e:
|
| 56 |
-
print(f"Error extracting response: {e}")
|
| 57 |
# If extraction fails, return the raw output
|
| 58 |
return str(outputs[0]["generated_text"])
|
| 59 |
|
|
|
|
| 43 |
|
| 44 |
outputs = pipe(messages, max_new_tokens=max_length)
|
| 45 |
|
| 46 |
+
# print("Raw output:", outputs) #removed after debugging output format problem is done
|
| 47 |
# Extract just the assistant's response
|
| 48 |
try:
|
| 49 |
# outputs[0]["generated_text"] is already a list of messages
|
|
|
|
| 53 |
if assistant_message["role"] == "assistant":
|
| 54 |
return assistant_message["content"]
|
| 55 |
except Exception as e:
|
| 56 |
+
# print(f"Error extracting response: {e}")
|
| 57 |
# If extraction fails, return the raw output
|
| 58 |
return str(outputs[0]["generated_text"])
|
| 59 |
|