Pyefuri commited on
Commit
b65edd1
·
verified ·
1 Parent(s): 140aae2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -16
app.py CHANGED
@@ -9,9 +9,10 @@ print("Memuat tokenizer...")
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)
10
 
11
  print("Memuat model Bahasa Biak...")
 
12
  model = AutoModelForCausalLM.from_pretrained(
13
  model_name,
14
- torch_dtype=torch.float16,
15
  device_map="auto"
16
  )
17
 
@@ -81,25 +82,31 @@ def prediksi_terjemahan(message, history):
81
  if next_token.item() == generation_kwargs["eos_token_id"]:
82
  break
83
 
84
- # ========================================================
85
- # PERBAIKAN UI/UX: MENGGUNAKAN ID UTAMA YANG DIJAMIN AMAN
86
- # ========================================================
87
  desain_kustom_css = """
88
- #wadah-chat {
89
- max-width: 800px !important; /* Membatasi lebar maksimal agar tidak melar ke samping */
90
- margin: 0 auto !important; /* Memaksa posisi kotak chat berada tepat di tengah monitor */
 
91
  }
92
  """
93
 
94
- # Menyusun interface dengan gr.Blocks standar yang didukung penuh oleh Hugging Face
95
- with gr.Blocks(theme=gr.themes.Soft(), css=desain_kustom_css) as demo:
96
- gr.Markdown("# 🤖 AI Bahasa Indonesia-Biak", elem_id="judul")
97
 
98
- # Perbaikan: Menggunakan gr.Column standar dengan elem_id langsung pada ChatInterface
99
- gr.ChatInterface(
100
- fn=prediksi_terjemahan,
101
- elem_id="wadah-chat" # Menyuntikkan CSS pembatas lebar langsung ke komponen utama
102
- )
 
103
 
104
  if __name__ == "__main__":
105
- demo.launch()
 
 
 
 
 
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)
10
 
11
  print("Memuat model Bahasa Biak...")
12
+ # Mengembalikan argumen ke 'dtype' sesuai warning log transformers Anda agar tidak deprecated
13
  model = AutoModelForCausalLM.from_pretrained(
14
  model_name,
15
+ dtype=torch.float16,
16
  device_map="auto"
17
  )
18
 
 
82
  if next_token.item() == generation_kwargs["eos_token_id"]:
83
  break
84
 
85
+ # ==========================================
86
+ # STRUKTUR DESAIN CUSTOM CSS
87
+ # ==========================================
88
  desain_kustom_css = """
89
+ #wadah-lebar-chat {
90
+ max-width: 800px !important; /* Membatasi lebar agar pas seperti ChatGPT */
91
+ margin: 0 auto !important; /* Menjaga posisi tetap presisi di tengah layar */
92
+ float: none !important;
93
  }
94
  """
95
 
96
+ # Membuka Blocks tanpa menyelipkan theme/css langsung di sini agar lolos dari Warning Gradio
97
+ with gr.Blocks() as demo:
98
+ gr.Markdown("# 🤖 AI Bahasa Indonesia-Biak")
99
 
100
+ # Solusi: Membungkus ChatInterface ke dalam gr.Column yang mendukung penuh elem_id
101
+ with gr.Column(elem_id="wadah-lebar-chat"):
102
+ gr.ChatInterface(
103
+ fn=prediksi_terjemahan,
104
+
105
+ )
106
 
107
  if __name__ == "__main__":
108
+ # PERBAIKAN: Memindahkan pengaturan tema dan CSS ke fungsi .launch() sesuai standarisasi sistem Gradio
109
+ demo.launch(
110
+ theme=gr.themes.Soft(),
111
+ css=desain_kustom_css
112
+ )