Update app.py

app.py

@@ -1,21 +1,23 @@
 import gradio as gr
-from transformers import GPT2Tokenizer, GPT2Model
+from transformers import pipeline, set_seed
 
-# Memuat tokenizer dan model GPT-2
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model = GPT2Model.from_pretrained('gpt2')
+# Memuat model GPT-2 untuk generasi teks
+generator = pipeline('text-generation', model='gpt2')
 
-# Fungsi untuk ekstraksi fitur dari teks
-def extract_features(text):
-    # Tokenisasi input
-    encoded_input = tokenizer(text, return_tensors='pt')
-    # Mendapatkan output (misalnya, hidden states)
-    output = model(**encoded_input)
-    # Menampilkan fitur yang diinginkan (misalnya, hidden states)
-    return output.last_hidden_state
+# Menetapkan seed untuk hasil yang konsisten
+set_seed(42)
+
+# Fungsi untuk menghasilkan respons chatbot
+def chatbot_response(prompt):
+    response = generator(prompt, max_length=50, num_return_sequences=1)
+    return response[0]['generated_text']
 
 # Membuat antarmuka Gradio untuk input dan output
-interface = gr.Interface(fn=extract_features, inputs="text", outputs="text")
+interface = gr.Interface(fn=chatbot_response, 
+                         inputs="text", 
+                         outputs="text",
+                         title="Chatbot with GPT-2", 
+                         description="Masukkan teks Anda, dan chatbot akan memberikan respons berdasarkan model GPT-2.")
 
 # Meluncurkan aplikasi
 interface.launch()