tomkr000 commited on
Commit
463cbdb
1 Parent(s): 63ccdfb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,5 +1,8 @@
1
  import gradio as gr
2
 
 
 
 
3
  def chat(message, history):
4
  new_user_input_ids = tokenizer.encode(message + tokenizer.eos_token, return_tensors='pt')
5
  bot_input_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1) if step > 0 else new_user_input_ids
 
1
  import gradio as gr
2
 
3
+ tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-small', padding_side='right')
4
+ model = AutoModelWithLMHead.from_pretrained('tomkr000/scottbotai')
5
+
6
  def chat(message, history):
7
  new_user_input_ids = tokenizer.encode(message + tokenizer.eos_token, return_tensors='pt')
8
  bot_input_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1) if step > 0 else new_user_input_ids