Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
|
|
5 |
pipe = pipeline("text-generation", model="unsloth/Llama-3.2-1B-Instruct")
|
6 |
|
7 |
def generate_text(prompt):
|
8 |
-
response = pipe(prompt, max_length=
|
9 |
return response[0]['generated_text']
|
10 |
|
11 |
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text",
|
|
|
5 |
pipe = pipeline("text-generation", model="unsloth/Llama-3.2-1B-Instruct")
|
6 |
|
7 |
def generate_text(prompt):
|
8 |
+
response = pipe(prompt, max_length=512, num_return_sequences=3)
|
9 |
return response[0]['generated_text']
|
10 |
|
11 |
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text",
|