Hugging Face's logo

Spaces:
mandar100
/
chatbot_dialogpt
Runtime error

App Files Community
125

Update app.py

#101
by taining - opened
base: refs/heads/main
←
from: refs/pr/101
Discussion Files changed
+4
-5
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,9 +1,8 @@
1
- from transformers import AutoModelForCausalLM, AutoTokenizer
2
- import torch
3
- import gradio as gr
4
 
5
- tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
6
- model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
7
 
8
  def predict(input, history=[]):
9
  # tokenize the new input sentence
 
1
+ from transformers import AutoTokenizer, AutoModelForMaskedLM
 
 
2
 
3
+ tokenizer=AutoTokenizer.from_pretrained("bert-base-chinese")
4
+
5
+ model=AutoModelForMaskedLM.from_pretrained("bert-base-chinese")
6
 
7
  def predict(input, history=[]):
8
  # tokenize the new input sentence