Spaces:

mandar100
/

chatbot_dialogpt

Runtime error

taining commited on Dec 24, 2022

Commit

ad1a716

1 Parent(s): df5b2b0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,8 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
-import gradio as gr
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 def predict(input, history=[]):
     # tokenize the new input sentence

+from transformers import AutoTokenizer, AutoModelForMaskedLM
+tokenizer = AutoTokenizer.from_pretrained("bert-base-chinese")
+model = AutoModelForMaskedLM.from_pretrained("bert-base-chinese")
 def predict(input, history=[]):
     # tokenize the new input sentence