Spaces:

sarves
/

bot01

Sleeping

sarves commited on Jul 11, 2023

Commit

394f51e

•

1 Parent(s): c2aa233

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,13 +5,13 @@ import gradio as gr
 import torch
-title = "🤖AI ChatBot"
 description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT)"
 examples = [["How are you?"]]
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 def predict(input, history=[]):

 import torch
+title = "IEEE CIS Summer School - 2023"
 description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT)"
 examples = [["How are you?"]]
+tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+model = AutoModelForCausalLM.from_pretrained("facebook/blenderbot-400M-distill")
 def predict(input, history=[]):