Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 
 @st.cache_resource(show_spinner=False)
 def load_model():
-    model_name = "Alijeff1214/DeutscheLexAI_BGB"
+    model_name = "Alijeff1214/DeutscheLexAI_BGB_2.0"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)
     return tokenizer, model
@@ -18,6 +18,6 @@ user_input = st.text_input("Enter your question or prompt:")
 if st.button("Generate Response") and user_input:
     # Tokenize and generate response (adjust parameters as needed)
     inputs = tokenizer(user_input, return_tensors="pt")
-    outputs = model.generate(**inputs, max_length=200, do_sample=True, temperature=0.7)
+    outputs = model.generate(**inputs, max_length=500, do_sample=True, temperature=0.7)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     st.text_area("Model Response:", value=response, height=300)