Update app.py

app.py

@@ -11,7 +11,6 @@ import torch
 import os
 import logging
 
-# Ensure proper display for debugging
 pd.set_option('display.max_colwidth', 1000)
 
 # Patch torch.load to always load on CPU
@@ -20,11 +19,10 @@ def cpu_load(*args, **kwargs):
     return original_torch_load(*args, map_location=torch.device('cpu'), **kwargs)
 torch.load = cpu_load
 
-# Flask app setup
 app = Flask(__name__)
 
 # Logging setup
-LOG_DIR = "/tmp/logs"  # Use a universally writable directory
+LOG_DIR = "/logs"  # Use a universally writable directory
 LOG_FILE = os.path.join(LOG_DIR, "usage_log.jsonl")
 os.makedirs(LOG_DIR, exist_ok=True)
 logging.basicConfig(
@@ -33,7 +31,6 @@ logging.basicConfig(
     format='%(asctime)s [%(levelname)s] %(message)s'
 )
 
-# Define pipelines
 PIPELINES = [
     {'id': 8, 'name': 'Embedded using BioWordVec', 'filename': "pipeline_ex3_s4.joblib"},
     {'id': 1, 'name': 'Baseline', 'filename': "pipeline_ex1_s1.joblib"},
@@ -47,7 +44,6 @@ PIPELINES = [
 
 pipeline_metadata = [{'id': p['id'], 'name': p['name']} for p in PIPELINES]
 
-# Helper functions
 def load_pipeline_from_hub(filename):
     cache_dir = "/tmp/hf_cache"
     os.environ["HF_HUB_CACHE"] = cache_dir
@@ -80,7 +76,7 @@ def log_interaction(user_input, model_name, predictions):
     except Exception as e:
         print(f"[ERROR] Could not write log entry: {e}")
 
-# Routes
+
 @app.route('/')
 def index():
     return render_template('index.html', pipelines=pipeline_metadata)
@@ -103,6 +99,6 @@ def get_data():
 
         return render_template('index.html', results=results, name=name, pipelines=pipeline_metadata)
 
-# Run the app
+
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860)