Update app.py

app.py

@@ -8,8 +8,11 @@ import os
 load_dotenv()
 
 # Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b")
-model = AutoModelForCausalLM.from_pretrained("google/gemma-2b")
+#tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b")
+#model = AutoModelForCausalLM.from_pretrained("google/gemma-2b")
+
+tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-rw-1b", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-rw-1b", trust_remote_code=True)
 
 # Function to generate blog content
 def generate_blog(topic, keywords):
@@ -26,7 +29,7 @@ def generate_blog(topic, keywords):
     Blog:
     """
     input_ids = tokenizer(prompt_template, return_tensors="pt", max_length=512, truncation=True)
-    outputs = model.generate(input_ids["input_ids"], max_length=800, num_return_sequences=1, temperature=0.7)
+    outputs = model.generate(input_ids["input_ids"], max_length=800, num_return_sequences=1)
     blog_content = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
     return blog_content