Update app.py
Browse files
app.py
CHANGED
@@ -4,12 +4,15 @@ import time
|
|
4 |
import gradio as gr
|
5 |
|
6 |
# Load a pre-trained chatbot model
|
7 |
-
chatbot = pipeline('text-generation', model='gpt2-large', max_length=
|
8 |
|
9 |
def wrap(x):
|
10 |
return textwrap.fill(x, replace_whitespace=False, fix_sentence_endings=True)
|
11 |
|
12 |
-
def chat(
|
13 |
-
|
|
|
|
|
|
|
14 |
|
15 |
gr.Interface(fn=chat, inputs="text", outputs="text").launch()
|
|
|
4 |
import gradio as gr
|
5 |
|
6 |
# Load a pre-trained chatbot model
|
7 |
+
chatbot = pipeline('text-generation', model='gpt2-large', max_length=200)
|
8 |
|
9 |
def wrap(x):
|
10 |
return textwrap.fill(x, replace_whitespace=False, fix_sentence_endings=True)
|
11 |
|
12 |
+
def chat(input_text, additional_param):
|
13 |
+
generated_text = chatbot(input_text)[0]["generated_text"]
|
14 |
+
print(f"Generated text: {generated_text}")
|
15 |
+
print(f"Additional parameter: {additional_param}")
|
16 |
+
return generated_text
|
17 |
|
18 |
gr.Interface(fn=chat, inputs="text", outputs="text").launch()
|