Spaces:
Running
on
Zero
Running
on
Zero
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 |
|