Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -117,7 +117,7 @@ class SinaReasonMedicalChat:
|
|
| 117 |
output = self.model.generate(**generation_kwargs)[0]
|
| 118 |
|
| 119 |
# Decode the response
|
| 120 |
-
full_response = self.tokenizer.decode(output[len(tokenized.input_ids) : (-1 if output[-1] == tokenizer.eos_token_id else len(output) ) ])
|
| 121 |
|
| 122 |
# Extract thinking and clinical summary
|
| 123 |
thinking, response = self.extract_thinking_and_response(full_response)
|
|
|
|
| 117 |
output = self.model.generate(**generation_kwargs)[0]
|
| 118 |
|
| 119 |
# Decode the response
|
| 120 |
+
full_response = self.tokenizer.decode(output[len(tokenized.input_ids) : (-1 if output[-1] == self.tokenizer.eos_token_id else len(output) ) ])
|
| 121 |
|
| 122 |
# Extract thinking and clinical summary
|
| 123 |
thinking, response = self.extract_thinking_and_response(full_response)
|