JohanBeytell commited on
Commit
ae3b9ed
·
verified ·
1 Parent(s): 8e6a5e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -6,11 +6,11 @@ from valx import detect_profanity, detect_hate_speech
6
  import gradio as gr
7
 
8
  sp = spm.SentencePieceProcessor()
9
- sp.Load("dungen_dev_preview.model")
10
 
11
- model = tf.keras.models.load_model("dungen_dev_preview_model.keras")
12
 
13
- max_seq_len = 25
14
 
15
  def generate_text(seed_text, next_words=30, temperature=0.5):
16
  seed_text = seed_text.strip().lower()
@@ -67,7 +67,7 @@ demo = gr.Interface(
67
  ],
68
  outputs=gr.Textbox(label="Generated Names"),
69
  title='Dungen Dev - Name Generator',
70
- description='A prompt-based name generator for game developers. Dungen Dev is an experimental model, and may produce outputs that are inappropriate, biased, or potentially harmful and inaccurate. Caution is advised.',
71
  examples=[
72
  ["a male character name", 30, 0.5],
73
  ["a futuristic city name", 30, 0.5],
 
6
  import gradio as gr
7
 
8
  sp = spm.SentencePieceProcessor()
9
+ sp.Load("dungen_dev_1.1.model")
10
 
11
+ model = tf.keras.models.load_model("dungen_dev_1.1.keras")
12
 
13
+ max_seq_len = 26
14
 
15
  def generate_text(seed_text, next_words=30, temperature=0.5):
16
  seed_text = seed_text.strip().lower()
 
67
  ],
68
  outputs=gr.Textbox(label="Generated Names"),
69
  title='Dungen Dev - Name Generator',
70
+ description='<sup>Dungen Dev v1.1</sup><br><br>A prompt-based name generator for game developers. Dungen Dev is an experimental model, and may produce outputs that are inappropriate, biased, or potentially harmful and inaccurate. Caution is advised.',
71
  examples=[
72
  ["a male character name", 30, 0.5],
73
  ["a futuristic city name", 30, 0.5],