Spaces:
Running
Running
Update app.py
Browse files
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(
|