Spaces:
Sleeping
Sleeping
cryptocalypse
commited on
Update gen.py
Browse files
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+" "+
|
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"])
|