rdlf commited on
Commit
6aa7c2f
·
verified ·
1 Parent(s): d8c99c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  tokenizer = GPT2Tokenizer.from_pretrained("sberbank-ai/mGPT")
6
  model = GPT2LMHeadModel.from_pretrained("sberbank-ai/mGPT")
7
 
8
- def eval_aguila(text):
9
  # Encode the input text
10
  input_ids = tokenizer.encode(text, return_tensors="pt")
11
 
@@ -29,6 +29,6 @@ def eval_aguila(text):
29
  return(f"Result: {generation[0]['generated_text']}")
30
 
31
 
32
- demo = gr.Interface(fn=lecturabilidad, inputs="text", outputs="text", title="Mixtral")
33
 
34
  demo.launch(share=True)
 
5
  tokenizer = GPT2Tokenizer.from_pretrained("sberbank-ai/mGPT")
6
  model = GPT2LMHeadModel.from_pretrained("sberbank-ai/mGPT")
7
 
8
+ def eval_text(text):
9
  # Encode the input text
10
  input_ids = tokenizer.encode(text, return_tensors="pt")
11
 
 
29
  return(f"Result: {generation[0]['generated_text']}")
30
 
31
 
32
+ demo = gr.Interface(fn=eval_text, inputs="text", outputs="text", title="Mixtral")
33
 
34
  demo.launch(share=True)