yasserrmd commited on
Commit
00657a2
·
verified ·
1 Parent(s): d45c7ca

Update app.py

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