Upload folder using huggingface_hub

book_generator.py

@@ -11,9 +11,9 @@ os.environ["GROQ_API_KEY"] = "gsk_iNUzpRQsiVfo4wZqvk2vWGdyb3FYm9VeDZSjUx1Jv3uOHQ
 
 # Step 4: Define helper functions for structured book generation
 def create_book_agent(
-    model_name: str = "llama3-70b-8192",
+    model_name: str = "llama-3.1-8b-instant",  # Updated model name
     temperature: float = 0.7,
-    max_tokens: int = 8192,
+    max_tokens: int = 16384,  # Increased token limit
     **kwargs
 ) -> ChatGroq:
     """Create a LangChain agent for book writing."""