Spaces:
Runtime error
Runtime error
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"])
|