Update app.py

app.py

@@ -21,6 +21,7 @@ classifier = pipeline("text-classification", model="distilbert/distilbert-base-u
 generator = pipeline("text2text-generation", model="google/flan-t5-base")
 
 # Function to classify customer comments
+@spaces.GPU
 def classify_comments(categories):
     global df  # Ensure we're modifying the global DataFrame
     sentiments = []
@@ -50,6 +51,7 @@ def classify_comments(categories):
     return df[['customer_id', 'customer_comment', 'comment_sentiment', 'comment_category', 'customer_nps', 'customer_segment']].to_html(index=False)
 
 # Function to generate visualizations
+@spaces.GPU
 def visualize_output():
     global df