Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
|
|
4 |
def generate_response(input_text, temperature, top_p):
|
5 |
from transformers import pipeline
|
6 |
# Carregar o modelo da Hugging Face localmente ou via API
|
7 |
-
model = pipeline("text-generation", model="
|
8 |
result = model(input_text, max_length=512, temperature=temperature, top_p=top_p)
|
9 |
return result[0]['generated_text']
|
10 |
|
@@ -14,8 +14,8 @@ with gr.Blocks() as demo:
|
|
14 |
with gr.Row():
|
15 |
input_text = gr.Textbox(label="Texto de entrada")
|
16 |
with gr.Row():
|
17 |
-
temperature = gr.Slider(0.1,
|
18 |
-
top_p = gr.Slider(0.1,
|
19 |
with gr.Row():
|
20 |
output_text = gr.Textbox(label="Texto gerado", interactive=False)
|
21 |
generate_button = gr.Button("Gerar Resposta")
|
|
|
4 |
def generate_response(input_text, temperature, top_p):
|
5 |
from transformers import pipeline
|
6 |
# Carregar o modelo da Hugging Face localmente ou via API
|
7 |
+
model = pipeline("text-generation", model="EmTpro01/CodeLlama-7b-finetuned-16bit")
|
8 |
result = model(input_text, max_length=512, temperature=temperature, top_p=top_p)
|
9 |
return result[0]['generated_text']
|
10 |
|
|
|
14 |
with gr.Row():
|
15 |
input_text = gr.Textbox(label="Texto de entrada")
|
16 |
with gr.Row():
|
17 |
+
temperature = gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperatura")
|
18 |
+
top_p = gr.Slider(0.1, 2.0, value=0.9, step=0.1, label="Top-p")
|
19 |
with gr.Row():
|
20 |
output_text = gr.Textbox(label="Texto gerado", interactive=False)
|
21 |
generate_button = gr.Button("Gerar Resposta")
|