Upload app.py with huggingface_hub

app.py

@@ -9,16 +9,38 @@ generator = None
 def initialize_model():
     global generator
     try:
-        # Use a reliable text generation model
+        # Use a modern instruction-tuned model for better document generation
         generator = pipeline(
             "text-generation",
-            model="distilgpt2",  # Smaller, faster model
+            model="microsoft/phi-2",  # Small but very capable instruction model
             device=-1,  # CPU only for compatibility
-            pad_token_id=50256  # Set pad token to avoid warnings
+            torch_dtype=torch.float32,
+            trust_remote_code=True
         )
         return "Model loaded successfully!"
     except Exception as e:
-        return f"Error loading model: {str(e)}"
+        # Fallback to Flan-T5 for instruction following
+        try:
+            from transformers import T5ForConditionalGeneration, T5Tokenizer
+            generator = pipeline(
+                "text2text-generation",
+                model="google/flan-t5-base",
+                device=-1,
+                torch_dtype=torch.float32
+            )
+            return "Fallback model (Flan-T5-base) loaded successfully!"
+        except Exception as e2:
+            # Final fallback to GPT2-medium
+            try:
+                generator = pipeline(
+                    "text-generation", 
+                    model="gpt2-medium",
+                    device=-1,
+                    torch_dtype=torch.float32
+                )
+                return "Final fallback model (GPT2-medium) loaded successfully!"
+            except Exception as e3:
+                return f"All models failed: Primary: {str(e)}, Secondary: {str(e2)}, Tertiary: {str(e3)}"
 
 def generate_onepager(topic, target_audience, key_points, tone, length):
     if generator is None:
@@ -28,20 +50,24 @@ def generate_onepager(topic, target_audience, key_points, tone, length):
     length_tokens = {"Short": 200, "Medium": 400, "Long": 600}
     max_tokens = length_tokens.get(length, 400)
     
-    prompt = f"""ONE-PAGE DOCUMENT
+    # Create a better prompt for modern instruction-following models
+    prompt = f"""Write a professional one-page document about "{topic}" for {target_audience}. 
 
-TOPIC: {topic}
-AUDIENCE: {target_audience}
-TONE: {tone}
+Use a {tone.lower()} tone and include these key points: {key_points}
 
-EXECUTIVE SUMMARY:
-{key_points}
+Format the document with:
+- Clear title
+- Executive summary
+- Main sections with headers
+- Key benefits/recommendations
+- Conclusion
 
-DETAILED CONTENT:
+Document:
 
-Title: {topic}
+# {topic}
 
-Overview:"""
+## Executive Summary
+"""
     
     try:
         # Generate the one-pager