william4416 commited on
Commit
743067f
1 Parent(s): 85d3e5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -1,4 +1,13 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 
 
 
 
 
 
 
2
 
3
  tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
4
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
+ import gradio as gr
3
+ import torch
4
+ import json
5
+
6
+
7
+ title = "????AI ChatBot"
8
+ description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT)"
9
+ examples = [["How are you?"]]
10
+
11
 
12
  tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
13
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")