Update app.py
Browse files
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
|
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=
|
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)
|