Update Spaces

app.py

@@ -4,7 +4,6 @@ from nltk.stem.porter import PorterStemmer
 import json
 import torch
 import torch.nn as nn
-from torch.utils.data import Dataset, DataLoader
 import random
 import gradio as gr
 
@@ -82,7 +81,12 @@ def predict(sentence):
         return "Saya tidak mengerti pertanyaan Anda."
 
 # Buat interface Gradio
-iface = gr.Interface(fn=predict, inputs="text", outputs="text", title="Chatbot Hukum Pajak")
+iface = gr.Interface(
+    fn=predict, 
+    inputs=gr.Textbox(placeholder="Silahkan masukkan pertanyaan...."), 
+    outputs="markdown", 
+    title="Chatbot Hukum Pajak",
+    description="Selamat Datang di chatbot pajak.\n\nChatbot ini bisa hanya bisa menjawab beberapa pertanyaan dasar tentang hukum pajak dari pasal 21, pasal 22, pasal 23, dan pasal 24.\n\nContoh pertanyaan seperti berikut:\n\n1. Apa itu pajak?\n\n2. Apa itu pajak penghasilan?\n\n3. Bisakah kamu jelaskan apa itu pajak penghasilan berdasarkan pasal 22?\n\n4. dan lain-lain.\n\nJika ingin melihat bagaimana chatbot ini dilatih, kalian bisa melihat dataset sederhana yang saya buat:\n\nhttps://huggingface.co/chandra10/chatbot_pajak/resolve/main/dataset.json\n\nModel chatbot ini belum sempurna. Jadi, silahkan beri saran di community")
 
 # Jalankan interface
 iface.launch(share=True)