NanoProgramerGamer commited on
Commit
cebaf80
·
verified ·
1 Parent(s): 2685982

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -1,20 +1,20 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Usa el modelo público y mantenido
5
- explain = pipeline("text2text-generation", model="Salesforce/codet5-base")
6
 
7
  def explicar_codigo(code):
8
- prompt = f"summarize: {code}"
9
- output = explain(prompt, max_length=128, do_sample=False)
10
  return output[0]["generated_text"]
11
 
12
  interface = gr.Interface(
13
  fn=explicar_codigo,
14
- inputs=gr.Textbox(lines=10, placeholder="Pega tu código aquí..."),
15
  outputs="text",
16
- title="Explicador de código con IA 🤖",
17
- description="Usa un modelo de Hugging Face para generar una explicación resumida del código que pegues.",
18
  theme="default"
19
  )
20
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Cambia aquí por un modelo más avanzado
5
+ explain = pipeline("text2text-generation", model="describeai/gemini-small")
6
 
7
  def explicar_codigo(code):
8
+ prompt = f"Explica detalladamente qué hace este fragmento de código Python:\n\n{code}"
9
+ output = explain(prompt, max_length=256, do_sample=False)
10
  return output[0]["generated_text"]
11
 
12
  interface = gr.Interface(
13
  fn=explicar_codigo,
14
+ inputs=gr.Textbox(lines=15, placeholder="Pega el código Python que quieras entender..."),
15
  outputs="text",
16
+ title="Asistente IA para explicar código Python",
17
+ description="Pega código Python y este modelo avanzado lo explicará en lenguaje natural.",
18
  theme="default"
19
  )
20