Spaces:
Sleeping
Sleeping
update
Browse files
app.py
CHANGED
@@ -6,9 +6,8 @@ import gradio as gr
|
|
6 |
model_name = "microsoft/DialoGPT-medium"
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
8 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
9 |
-
llm_config = {"model": model, "tokenizer": tokenizer}
|
10 |
|
11 |
-
agent = ConversationChain(
|
12 |
|
13 |
# Define the Gradio interface
|
14 |
def chatbot_interface(input_text):
|
|
|
6 |
model_name = "microsoft/DialoGPT-medium"
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
8 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
|
|
9 |
|
10 |
+
agent = ConversationChain(model=model, tokenizer=tokenizer)
|
11 |
|
12 |
# Define the Gradio interface
|
13 |
def chatbot_interface(input_text):
|