Update app.py

app.py

@@ -6,9 +6,21 @@ from huggingface_hub import login
 
 
 login(token = os.getenv('HF_TOKEN'))
-pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", torch_dtype=torch.bfloat16)
+pipe = pipeline("text-generation", model="google/gemma-3-1b-it", torch_dtype=torch.bfloat16)
 
 txt = st.text_area("Test this")
 
 if (txt):
-    st.write(pipe(txt, max_new_tokens=50))
+    messages = [
+        [
+            {
+                "role": "system",
+                "content": [{"type": "text", "text": "You are a helpful assistant."},]
+            },
+            {
+                "role": "user",
+                "content": [{"type": "text", "text": "Write a poem on Hugging Face, the company"},]
+            },
+        ],
+    ]
+    st.write(pipe(messages, max_new_tokens=50))