Update app.py

app.py

@@ -19,7 +19,7 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
-@spaces.GPU
+@spaces.GPU(duration=4*60)
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],
@@ -64,9 +64,10 @@ def generate(
 chat_interface = gr.ChatInterface(
     fn=generate,
     chatbot=gr.Chatbot(height=450,
-                      label="google/gemma-2-27b-it",
+                      label="google/gemma-2-9b-it",
                       show_share_button=True,
-                      avatar_images=(None, 'geitje-logo.jpg')),
+                      # avatar_images=(None, 'geitje-logo.jpg')
+                      ),
     # textbox=gr.Textbox(value="Typ een bericht…"),
     cache_examples=False,
     additional_inputs=[
@@ -134,8 +135,8 @@ Twee van de wortelbedrijven werkten mee door meer informatie over de ongeoorloof
 ```
 Vat bovenstaand artikel samen"""]
     ],
-    title="google/gemma-2-27b-it",
-    description="""google/gemma-2-27b-it quick demo""",
+    title="google/gemma-2-9b-it",
+    description="""google/gemma-2-9b-it quick demo""",
     submit_btn="Genereer",
     stop_btn="Stop",
     retry_btn="🔄 Opnieuw",