VuAI commited on
Commit
60be76c
·
1 Parent(s): 0358238

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,6 +3,8 @@ import gradio as gr
3
  import supabase
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
 
 
 
6
  # Initialize the Supabase client
7
  supabase_url = os.getenv("supabase_url")
8
  supabase_token = os.getenv("supabase_token")
@@ -65,7 +67,7 @@ def login(username, password):
65
  data = client.auth.sign_in_with_password({"email": username, "password": password})
66
  return data
67
 
68
- MODELS = ['VuAI/khi-van', 'VuAI/vi2vi_vn98']
69
 
70
  curr_words = 9999
71
  with gr.Blocks(title="VN98 Translator", theme=gr.themes.Default(primary_hue="red", secondary_hue="red")) as demo:
 
3
  import supabase
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
 
6
+ MODELS = ['VuAI/khi-van', 'VuAI/tuan', 'VuAI/vi2vi_vn98']
7
+
8
  # Initialize the Supabase client
9
  supabase_url = os.getenv("supabase_url")
10
  supabase_token = os.getenv("supabase_token")
 
67
  data = client.auth.sign_in_with_password({"email": username, "password": password})
68
  return data
69
 
70
+
71
 
72
  curr_words = 9999
73
  with gr.Blocks(title="VN98 Translator", theme=gr.themes.Default(primary_hue="red", secondary_hue="red")) as demo: