update

app.py

@@ -6,6 +6,7 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 # from averitec.models.AveritecModule import Wikipediaretriever, Googleretriever, veracity_prediction, justification_generation
 import uvicorn
+import spaces
 
 app = FastAPI()
 
@@ -259,7 +260,7 @@ class SequenceClassificationDataLoader(pl.LightningDataModule):
                 + bool_explanation
         )
 
-
+@spaces.GPU
 def veracity_prediction(claim, evidence):
     dataLoader = SequenceClassificationDataLoader(
         tokenizer=veracity_tokenizer,
@@ -335,7 +336,7 @@ def extract_claim_str(claim, evidence, verdict_label):
 
     return claim_str
 
-
+@spaces.GPU
 def justification_generation(claim, evidence, verdict_label):
     #
     claim_str = extract_claim_str(claim, evidence, verdict_label)
@@ -362,12 +363,13 @@ def log_on_azure(file, logs, azure_share_client):
     file_client.upload_file(logs)
 
 
+@spaces.GPU
 @app.post("/predict/")
 def fact_checking(item: Item):
-    # claim = item['claim']
-    # source = item['source']
-    claim = item.claim
-    source = item.source
+    claim = item['claim']
+    source = item['source']
+    # claim = item.claim
+    # source = item.source
 
     # Step1: Evidence Retrieval
     if source == "Wikipedia":