PayPeer commited on
Commit
053a38c
1 Parent(s): f877950

The 'doctor' is now a 'static' method.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -368,8 +368,9 @@ peft_model = PeftModel.from_pretrained(peft_model, "Tonic/stablemed", token=hf_t
368
  class ChatBot:
369
  def __init__(self):
370
  self.history = []
371
-
372
- def doctor(self, user_input, system_prompt="You are an expert medical analyst:"):
 
373
  formatted_input = f"{system_prompt}{user_input}"
374
  user_input_ids = tokenizer.encode(formatted_input, return_tensors="pt")
375
  response = peft_model.generate(input_ids=user_input_ids, max_length=512, pad_token_id=tokenizer.eos_token_id)
 
368
  class ChatBot:
369
  def __init__(self):
370
  self.history = []
371
+
372
+ @staticmethod
373
+ def doctor(user_input, system_prompt="You are an expert medical analyst:"):
374
  formatted_input = f"{system_prompt}{user_input}"
375
  user_input_ids = tokenizer.encode(formatted_input, return_tensors="pt")
376
  response = peft_model.generate(input_ids=user_input_ids, max_length=512, pad_token_id=tokenizer.eos_token_id)