luelhagos commited on
Commit
fa1cc91
Β·
1 Parent(s): e8130f9

model updated

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline, GPT2TokenizerFast
3
 
4
- modelId = "luel/gpt2-tigrinya-small"
5
 
6
  tokenizer = GPT2TokenizerFast.from_pretrained(modelId, model_max_length=128)
7
 
@@ -71,8 +71,8 @@ def create_interface():
71
 
72
  gr.Examples(
73
  examples=[
74
- ["α‰΅αŒαˆ«α‹­"],
75
- ["αŠ£α‹²αˆ΅ αŠ£α‰ α‰£"],
76
  ["αˆ°αˆ‹αˆ"]
77
  ],
78
  inputs=input_prompt
 
1
  import gradio as gr
2
  from transformers import pipeline, GPT2TokenizerFast
3
 
4
+ modelId = "luel/gpt2-tigrinya-medium"
5
 
6
  tokenizer = GPT2TokenizerFast.from_pretrained(modelId, model_max_length=128)
7
 
 
71
 
72
  gr.Examples(
73
  examples=[
74
+ ["ክልል α‰΅αŒαˆ«α‹­"],
75
+ ["መረጻ ኣሜαˆͺካ"],
76
  ["αˆ°αˆ‹αˆ"]
77
  ],
78
  inputs=input_prompt
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  transformers==4.31.0
2
- torch==2.5.0
 
 
1
  transformers==4.31.0
2
+ torch==2.5.0
3
+ xformers