Update app.py

app.py

@@ -17,7 +17,7 @@ model.to(device)
 
 def generate_code(prompt):
     # Refined prompt
-    full_prompt = f"Please generate a basic HTML template for a personal blog. The output should be only the HTML code."
+    full_prompt = f"Please generate a basic HTML template for a personal blog, including a header, main content area for posts, and a footer. The output should be only the HTML code."
     
     # Tokenize the input
     input_tensor = tokenizer(full_prompt, return_tensors="pt", padding=True, truncation=True).to(device)