GPU initialized

app.py

@@ -54,7 +54,7 @@ processor = AutoProcessor.from_pretrained(model_name, use_auth_token=HF_TOKEN)
 #     # Decode the output to return the final response
 #     response = processor.decode(outputs[0], skip_special_tokens=True)
 #     return response
-# @spaces.GPU
+@spaces.GPU
 def predict_text(text):
     # Prepare the input messages
     messages = [{"role": "user", "content": [{"type": "text", "text": text}]}]