Update app.py

app.py

@@ -166,7 +166,7 @@ def generate(
     model.to(device)
     input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
     generated_ids = model.generate(**input_ids, **generate_kwargs)
-    return tokenizer.decode(generated_ids[0][input_ids["input_ids"].shape[1]:], skip_special_tokens=True).strip()
+    return tokenizer.decode(generated_ids[0][input_ids["input_ids"].shape[1]:], skip_special_tokens=True)
 
 
 examples = [
@@ -222,7 +222,7 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=css) as demo:
         with gr.Row():
             library = gr.Dropdown(
                         ["SQLModel", "SfePy", "MegEngine", "LangChain", "LlamaIndex", "DSPy"],
-                        value="LangChain",
+                        value="LlamaIndex",
                         label="Library",
                         info="Choose a library from the list",
                         )