Update app.py

app.py

@@ -9,6 +9,7 @@ from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import List, Optional, Dict, Any
+import spaces
 
 # Variabili globali
 model = None
@@ -163,6 +164,11 @@ def create_gradio_interface():
     
     return interface
 
+# Decorator per richiedere la GPU dallo space
+@spaces.GPU
+def get_gpu():
+    return "GPU allocata con successo"
+
 # Crea l'applicazione FastAPI
 app = FastAPI()
 
@@ -236,6 +242,8 @@ app = gr.mount_gradio_app(app, demo, path="/")
 # Precarica il modello all'avvio (usando il nuovo metodo lifespan invece di on_event)
 @app.on_event("startup")
 async def startup_load_model():
+    # Assicurati che la GPU sia allocata prima di caricare il modello
+    get_gpu()
     load_model()
 
 # Per Hugging Face Spaces, assicurati che l'app sia esportata correttamente