mateoluksenberg commited on
Commit
dc448fb
·
verified ·
1 Parent(s): 38d5aa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -209,6 +209,7 @@ EXAMPLES = [
209
 
210
 
211
  # Definir la función simple_chat
 
212
  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):
213
  # Cargar el modelo preentrenado
214
  model = AutoModelForCausalLM.from_pretrained(
 
209
 
210
 
211
  # Definir la función simple_chat
212
+ @spaces.GPU()
213
  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):
214
  # Cargar el modelo preentrenado
215
  model = AutoModelForCausalLM.from_pretrained(