Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|