kazimsayed commited on
Commit
6ff692c
·
1 Parent(s): 1cefe7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ tokenizer = AutoTokenizer.from_pretrained("docto/Docto-Bot")
5
  model = AutoModelForCausalLM.from_pretrained("docto/Docto-Bot")
6
  special_token = '<|endoftext|>'
7
 
8
- def get_reply(user_input):
9
- prompt_text = f'Question: {user_input}\nAnswer:'
10
  encoded_prompt = tokenizer.encode(prompt_text,
11
  add_special_tokens = False,
12
  return_tensors = 'pt')
 
5
  model = AutoModelForCausalLM.from_pretrained("docto/Docto-Bot")
6
  special_token = '<|endoftext|>'
7
 
8
+ def get_reply(userinput):
9
+ prompt_text = f'Question: {userinput}\nAnswer:'
10
  encoded_prompt = tokenizer.encode(prompt_text,
11
  add_special_tokens = False,
12
  return_tensors = 'pt')