Update app.py

app.py

@@ -34,6 +34,9 @@ def load_reddit_split(subreddit_name):
 # Combine subreddit data
 combined_dataset = chain(*(load_reddit_split(sub) for sub in target_subreddits))
 
+if "JAVA_HOME" not in os.environ:
+    os.environ["JAVA_HOME"] = "/usr/lib/jvm/java-11-openjdk-amd64"  # Common path on HF Spaces/Debian
+
 # PySpark session
 spark = SparkSession.builder.getOrCreate()
 df = spark.createDataFrame([{"body": ex["body"]} for ex in islice(combined_dataset, 100000)])