ricklegac commited on
Commit
aef5ab5
·
1 Parent(s): 8172579
Files changed (1) hide show
  1. app.py +2 -10
app.py CHANGED
@@ -1,15 +1,7 @@
1
- #!/usr/bin/env python3
2
- # -*- coding: utf-8 -*-
3
- """
4
- Created on Thu Apr 17 20:18:35 2025
5
-
6
- @author: rick
7
- """
8
-
9
  import gradio as gr
10
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
11
 
12
- model_id = "mistralai/Mistral-7B-Instruct-v0.1"
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_id)
15
  model = AutoModelForCausalLM.from_pretrained(
@@ -23,7 +15,7 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
23
  def generar_commit(diff):
24
  prompt = f"Escribí un mensaje de commit claro, corto y técnico basado en los siguientes cambios:\n{diff}"
25
  output = pipe(prompt, max_new_tokens=60, do_sample=True, temperature=0.7)
26
- return [output[0]["generated_text"]] # <-- esta línea es clave
27
 
28
  demo = gr.Interface(
29
  fn=generar_commit,
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
+ model_id = "HuggingFaceH4/zephyr-7b-alpha" # ✅ sin restricciones
5
 
6
  tokenizer = AutoTokenizer.from_pretrained(model_id)
7
  model = AutoModelForCausalLM.from_pretrained(
 
15
  def generar_commit(diff):
16
  prompt = f"Escribí un mensaje de commit claro, corto y técnico basado en los siguientes cambios:\n{diff}"
17
  output = pipe(prompt, max_new_tokens=60, do_sample=True, temperature=0.7)
18
+ return [output[0]["generated_text"]]
19
 
20
  demo = gr.Interface(
21
  fn=generar_commit,