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)
|