benhaotang commited on
Commit
91706fb
·
verified ·
1 Parent(s): 06d5510

remove debug note

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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