cryptocalypse commited on
Commit
278e21a
·
verified ·
1 Parent(s): c6ce46e

Update gen.py

Browse files
Files changed (1) hide show
  1. gen.py +2 -2
gen.py CHANGED
@@ -146,9 +146,9 @@ prompt = (
146
  )
147
 
148
 
149
- def generate():
150
  # Generar el texto usando el modelo
151
- outputs = pipe(prompt+" "+sys.argv[1], max_new_tokens=1024, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
152
 
153
  # Imprimir la salida generada
154
  print(outputs[0]["generated_text"])
 
146
  )
147
 
148
 
149
+ def generate(event):
150
  # Generar el texto usando el modelo
151
+ outputs = pipe(prompt+" "+event, max_new_tokens=1024, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
152
 
153
  # Imprimir la salida generada
154
  print(outputs[0]["generated_text"])