yasserrmd commited on
Commit
592c30f
·
verified ·
1 Parent(s): 711acf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ class SinaReasonMedicalChat:
97
  # Add current message
98
  messages.append({"role": "user", "content": message})
99
 
100
- tokenized = tokenizer.apply_chat_template(messages, return_dict=True)
101
 
102
  input_ids = torch.tensor(tokenized.input_ids, device="cuda").unsqueeze(0)
103
  attention_mask = torch.tensor(tokenized.attention_mask, device="cuda").unsqueeze(0)
 
97
  # Add current message
98
  messages.append({"role": "user", "content": message})
99
 
100
+ tokenized = self.tokenizer.apply_chat_template(messages, return_dict=True)
101
 
102
  input_ids = torch.tensor(tokenized.input_ids, device="cuda").unsqueeze(0)
103
  attention_mask = torch.tensor(tokenized.attention_mask, device="cuda").unsqueeze(0)