Update app.py

app.py

@@ -1,34 +1,23 @@
 import os
-from transformers import pipeline
 
-# ✅ Set Hugging Face token (replace with your token)
-# Option 1: Directly in script (works locally)
-os.environ["HUGGINGFACE_HUB_TOKEN"] = "YOUR_HUGGINGFACE_TOKEN_HERE"
+# ✅ Set your Hugging Face token before any Hugging Face imports
+os.environ["HUGGINGFACE_HUB_TOKEN"] = "YOUR_HUGGINGFACE_TOKEN"  # Replace with your actual token
 
-# Option 2: If running in a container, make sure HUGGINGFACE_HUB_TOKEN
-# is set as an environment variable externally (Docker, cloud, etc.)
+from transformers import pipeline
 
-# Model names
-MEDINI_MODEL = "PuruAI/Medini_Intelligence"
+# Model configuration
+MODEL_ID = "PuruAI/Medini_Intelligence"
 FALLBACK_MODEL = "gpt2"
 
-# Function to safely load a model
-def load_model(model_name, fallback_model):
-    try:
-        print(f"Loading model: {model_name}")
-        generator = pipeline("text-generation", model=model_name)
-        print("✅ Model loaded successfully!")
-        return generator
-    except Exception as e:
-        print(f"⚠️ Failed to load {model_name}: {e}")
-        print(f"⏩ Falling back to {fallback_model}")
-        generator = pipeline("text-generation", model=fallback_model)
-        return generator
-
-# Load Medini model or fallback
-generator = load_model(MEDINI_MODEL, FALLBACK_MODEL)
+# Try loading your main model
+try:
+    generator = pipeline("text-generation", model=MODEL_ID)
+    print("✅ Medini_AI loaded successfully!")
+except Exception as e:
+    print(f"⚠️ Failed to load Medini_AI: {e}")
+    print("⏩ Falling back to GPT-2...")
+    generator = pipeline("text-generation", model=FALLBACK_MODEL)
 
-# Example usage
-prompt = "Hello Medini AI, can you explain AI agents?"
-result = generator(prompt, max_length=100, do_sample=True)
+# Test the pipeline
+result = generator("Hello world!", max_length=50)
 print(result)