mateoluksenberg commited on
Commit
be9803e
·
verified ·
1 Parent(s): 8370975

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -208,7 +208,7 @@ EXAMPLES = [
208
 
209
 
210
  # Definir la función simple_chat
211
- def simple_chat(message: Message, 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,
 
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,