Spaces:

Yhhxhfh
/

dgdgdgdgd

Runtime error

Yhhxhfh commited on Sep 22

Commit

faca04b

•

1 Parent(s): fe0bd35

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ app = FastAPI()
 async def root():
     return {"message": "Modelo entrenado y en ejecución."}
-@spaces.gpu
 def load_and_train():
     model_name = 'gpt2'
     tokenizer = GPT2Tokenizer.from_pretrained(model_name)

 async def root():
     return {"message": "Modelo entrenado y en ejecución."}
+@spaces.GPU
 def load_and_train():
     model_name = 'gpt2'
     tokenizer = GPT2Tokenizer.from_pretrained(model_name)