Update app.py

app.py

@@ -25,6 +25,7 @@ def return_prediction(prompt):
     ans_pred = ans.split('### Response:')[1].split('<')[0]
     return ans_pred
 
+@spaces.GPU
 def classify_website(site_text):
     prompt = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.