Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -208,7 +208,7 @@ EXAMPLES = [
|
|
208 |
|
209 |
|
210 |
# Definir la función simple_chat
|
211 |
-
def simple_chat(message:
|
212 |
# Cargar el modelo preentrenado
|
213 |
model = AutoModelForCausalLM.from_pretrained(
|
214 |
MODEL_ID,
|
|
|
208 |
|
209 |
|
210 |
# Definir la función simple_chat
|
211 |
+
def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096, top_p: float = 1, top_k: int = 10, penalty: float = 1.0):
|
212 |
# Cargar el modelo preentrenado
|
213 |
model = AutoModelForCausalLM.from_pretrained(
|
214 |
MODEL_ID,
|