Update app.py

app.py

@@ -5,6 +5,8 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer, Trainer, TrainingArgume
 from datasets import load_dataset, concatenate_datasets
 from huggingface_hub import login
 import time
+import uvicorn
+from fastapi import FastAPI
 
 load_dotenv()
 login(token=os.getenv('HUGGINGFACE_TOKEN'))
@@ -53,6 +55,12 @@ trainer = Trainer(
     train_dataset=tokenized_dataset,
 )
 
+app = FastAPI()
+
+@app.get("/")
+async def root():
+    return {"message": "Modelo entrenado y en ejecución."}
+
 @spaces.gpu
 def run_training():
     while True:
@@ -65,7 +73,7 @@ def run_training():
             print(f"Error durante el entrenamiento: {e}. Reiniciando el proceso de entrenamiento...")
             time.sleep(10)
 
-run_training()
-
-import shutil
-shutil.rmtree('./results', ignore_errors=True)
+if __name__ == "__main__":
+    import threading
+    threading.Thread(target=run_training).start()
+    uvicorn.run(app, host="0.0.0.0", port=7860)