Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
 import os
 
-model_name = "google/gemma-4-E4B-it" 
+model_name = "google/gemma-4-E2B-it" 
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
@@ -14,7 +14,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name,
 pipe = pipeline("text-generation", 
                 model=model_name, 
                 tokenizer=tokenizer, 
-                max_new_tokens=150, 
+                max_new_tokens=1500, 
                 temperature=0.7)
 
 def generate_response(message, history):
@@ -38,8 +38,8 @@ def generate_response(message, history):
 
 demo = gr.ChatInterface(
     generate_response,
-    title="ELIZA (com LLM)",
-    description="Compartilhe seus pensamentos e ELIZA irá ajudar você a refletir sobre eles."
+    title="Brain map(com LLM)",
+    description="You are a Gemma 4 model, a trusted speaker and medical assistant. "
 )
 
 demo.launch()