Update app.py

app.py

@@ -6,8 +6,8 @@ import re
 import random
 
 # Initialize the model
-model = AutoModelForCausalLM.from_pretrained("Detsutut/Igea-1B-instruct-v0.2-GGUF", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
-tokenizer = AutoTokenizer.from_pretrained( "Detsutut/Igea-1B-instruct-v0.2")
+model = AutoModelForCausalLM.from_pretrained("Detsutut/Igea-1B-instruct-v0.3-test4epochs-GGUF", model_file="unsloth.Q4_K_M.gguf", model_type="mistral", hf=True)
+tokenizer = AutoTokenizer.from_pretrained( "Detsutut/Igea-1B-instruct-v0.3")
 
 
 gen_pipeline = pipeline(
@@ -35,7 +35,7 @@ alpaca_instruct_prompt = """{}
 {}"""
 
 # Define the function to generate text
-def generate_text(input_text, max_new_tokens=100, temperature=1, system_prompt=""):
+def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt=""):
 
     if len(system_prompt)>0:
         system_str = system_prompt