Update guide_model.py

guide_model.py

@@ -1,47 +1,29 @@
 import os
 
-# Set cache-related env vars BEFORE importing transformers
-cache_dir = "/tmp/hf_cache"
-os.environ["HF_HOME"] = cache_dir
-os.environ["TRANSFORMERS_CACHE"] = cache_dir
-os.environ["HF_DATASETS_CACHE"] = cache_dir
-os.environ["HF_METRICS_CACHE"] = cache_dir
-
-# Ensure cache directory exists; if permission denied, fallback to ./hf_cache
-try:
-    os.makedirs(cache_dir, exist_ok=True)
-except PermissionError:
-    cache_dir = "./hf_cache"
-    os.makedirs(cache_dir, exist_ok=True)
-    os.environ["HF_HOME"] = cache_dir
-    os.environ["TRANSFORMERS_CACHE"] = cache_dir
-    os.environ["HF_DATASETS_CACHE"] = cache_dir
-    os.environ["HF_METRICS_CACHE"] = cache_dir
+# Set cache environment variables BEFORE importing transformers
+os.environ["HF_HOME"] = "/tmp/hf_cache"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_DATASETS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_METRICS_CACHE"] = "/tmp/hf_cache"
 
+import transformers
 from transformers import pipeline
 
-# Load the text generation pipeline once
+os.makedirs("/tmp/hf_cache", exist_ok=True)
+
 try:
-    generator = pipeline(
-        "text-generation",
-        model="gpt2",
-        cache_dir=cache_dir
-    )
+    generator = pipeline("text-generation", model="gpt2", cache_dir="/tmp/hf_cache")
 except Exception as e:
     generator = None
     print("⚠️ Failed to load model:", e)
 
-
 def generate_description(country_name):
     if not generator:
         return "⚠️ Model is not available. Please check the server logs."
 
+    prompt = f"Tell me about {country_name}."
     try:
-        prompt = f"Tell me about {country_name}."
         result = generator(prompt, max_length=100, do_sample=True)
-        if result and isinstance(result, list):
-            return result[0]["generated_text"].strip()
-        else:
-            return "⚠️ Failed to generate text."
+        return result[0]["generated_text"].strip()
     except Exception as e:
         return f"⚠️ Error generating description: {str(e)}"