Update app.py

app.py

@@ -19,14 +19,12 @@ def chat(message, history, system_prompt):
         messages.append({"role": "user", "content": user})
         messages.append({"role": "assistant", "content": assistant})
     messages.append({"role": "user", "content": message})
-
     text = tokenizer.apply_chat_template(
         messages,
         tokenize=False,
         add_generation_prompt=True
     )
     inputs = tokenizer(text, return_tensors="pt").to(model.device)
-
     with torch.no_grad():
         outputs = model.generate(
             **inputs,
@@ -35,7 +33,6 @@ def chat(message, history, system_prompt):
             do_sample=True,
             pad_token_id=tokenizer.eos_token_id
         )
-
     response = tokenizer.decode(
         outputs[0][inputs["input_ids"].shape[1]:],
         skip_special_tokens=True
@@ -57,8 +54,7 @@ demo = gr.ChatInterface(
         ["Halo, siapa kamu?"],
         ["Produk saya belum sampai, tolong bantu"],
         ["Rangkum teks berikut: Snapgate adalah platform layanan pelanggan berbasis AI"]
-    ],
-    theme=gr.themes.Soft()
+    ]
 )
 
 demo.launch()