Thziin commited on
Commit
6fcde8c
·
verified ·
1 Parent(s): 5f6d0c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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="meta-llama/Llama-3.3-70B-Instruct")
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, 1.0, value=0.7, step=0.1, label="Temperatura")
18
- top_p = gr.Slider(0.1, 1.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")
 
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")