wahdie commited on
Commit
af9896b
·
1 Parent(s): a1bbba3

Add application file

Browse files
Files changed (2) hide show
  1. app.py +25 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr # Mengimpor pustaka Gradio untuk membuat UI
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM # Mengimpor tokenizer dan model dari HuggingFace
3
+
4
+ # 1. Load tokenizer dan model dari HuggingFace Hub
5
+ model_name = "wahdie11/text2py2" # Ganti dengan path ke model Anda di HuggingFace Hub
6
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = AutoModelForCausalLM.from_pretrained(model_name)
8
+
9
+ # 2. Definisikan fungsi untuk menghasilkan kode dari input teks
10
+ def generate_code(prompt):
11
+ inputs = tokenizer(prompt, return_tensors="pt") # Mengonversi prompt ke tensor
12
+ outputs = model.generate(**inputs, max_length=256) # Menghasilkan kode dengan panjang maksimum 256 token
13
+ return tokenizer.decode(outputs[0], skip_special_tokens=True) # Mendekode output menjadi string
14
+
15
+ # 3. Buat interface menggunakan Gradio
16
+ demo = gr.Interface(
17
+ fn=generate_code, # Fungsi yang akan dijalankan
18
+ inputs="text", # Input berupa teks
19
+ outputs="text", # Output berupa teks
20
+ title="Text-to-Code Generator", # Judul aplikasi
21
+ description="Masukkan deskripsi teks untuk menghasilkan kode." # Deskripsi aplikasi
22
+ )
23
+
24
+ # 4. Menjalankan interface
25
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ transformers>=4.0.0
2
+ torch
3
+ gradio