Update app.py

app.py

@@ -12,6 +12,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 from sentence_transformers import SentenceTransformer
 import torch
 import langchain
+import spaces
 
 app = FastAPI()
 
@@ -49,6 +50,7 @@ def read_root():
     return {"Hello": "World"}
 
 @app.post("/v1/generateText")
+@spaces.GPU()
 async def generateText(request: Request):
     request_dict = await request.json()
     prompt = request_dict.pop("prompt")