Update app.py

app.py

@@ -10,7 +10,7 @@ model = MultiHeadClassification.from_pretrained(
 ).to('cuda')
 tokenizer = AutoTokenizer.from_pretrained('BAAI/bge-m3')
 
-@spaces.GPU()
+#@spaces.GPU()
 def generate(prompt):
     inputs = tokenizer([prompt]).to('cuda')
     return model(**inputs)