model

app.py

@@ -2,10 +2,11 @@ import spaces
 import gradio as gr
 from transformers import pipeline, GPT2TokenizerFast
 
-mode_id = "alakxender/dv-wiki-gpt2"
-tokenizer = GPT2TokenizerFast.from_pretrained(mode_id, model_max_length=128)
+#model_id = "alakxender/dv-wiki-gpt2"
+model_id = "alakxender/dv-articles-gpt2"
+tokenizer = GPT2TokenizerFast.from_pretrained(model_id, model_max_length=128)
 
-generator = pipeline("text-generation", model=mode_id, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, eos_token_id=tokenizer.eos_token_id)
+generator = pipeline("text-generation", model=model_id, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, eos_token_id=tokenizer.eos_token_id)
 
 @spaces.GPU
 def generate_text(prompt, max_length, temperature):
@@ -38,7 +39,7 @@ def create_interface(css=styles):
     with gr.Blocks() as demo:
         gr.Markdown("# Dhivehi Text Generator (GPT-2, Wiki)")
         gr.Markdown(
-            "This is a GPT-2 model trained from Dhivehi text data from wikipedia"
+            "This is a GPT-2 model trained from Dhivehi text data from wikipedia\n"
             "Enter some text and generate a new text, adjust the parameters to generate text."
         )
         gr.Markdown("""