Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 text_generator = None
 is_hugging_face = True
 model_id = "google/gemma-2-9b-it"
+model_id = "google/gemma-2-2b-it"
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
 device = "cuda"
@@ -52,7 +53,7 @@ if not is_hugging_face:
 
 print("initialized")
 
-@spaces.GPU(duration=10)
+@spaces.GPU(duration=60)
 def generate_text(messages):
     if is_hugging_face:#need everytime initialize for ZeroGPU
         model = AutoModelForCausalLM.from_pretrained(