Update app.py

app.py

@@ -1,23 +1,21 @@
-from huggingface_hub import InferenceClient
-import os
+import gradio as gr
+from transformers import GPT2Tokenizer, GPT2Model
 
-# Ambil token dari Repository secrets
-api_token = os.getenv("LLM")
+# Memuat tokenizer dan model GPT-2
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
+model = GPT2Model.from_pretrained('gpt2')
 
-# Inisialisasi Inference Client
-client = InferenceClient(api_key=api_token)
+# 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
 
-# Pesan dari user
-messages = [
-    {"role": "user", "content": "What is the capital of France?"}
-]
+# Membuat antarmuka Gradio untuk input dan output
+interface = gr.Interface(fn=extract_features, inputs="text", outputs="text")
 
-# Inferensi menggunakan model
-completion = client.chat.completions.create(
-    model="meta-llama/Llama-3.3-70B-Instruct",
-    messages=messages,
-    max_tokens=500
-)
-
-# Cetak jawaban dari model
-print(completion.choices[0].message)
+# Meluncurkan aplikasi
+interface.launch()