File size: 835 Bytes
741577a
 
1017c2b
1cee461
741577a
 
1cee461
 
 
 
741577a
 
1017c2b
741577a
1017c2b
 
741577a
963f714
741577a
 
 
 
 
 
 
 
 
 
 
 
 
2ca0523
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import gradio as gr
from transformers import pipeline
import transformers
from functools import lru_cache


@lru_cache(maxsize=4)
def get_pipeline(task: str, model_url: str):
    print(f"Loading pipeline: task={task}, model={model_url}")
    return pipeline(task, model=model_url)

def translate_text(task, modelUrl, chunk_to_translate):
    print(transformers.__version__)

    modelUrl = modelUrl.strip()
    task = task.strip()

    pipe = get_pipeline(task, modelUrl)
    result = pipe(chunk_to_translate)
    print(f"translated chunk is: {result}")

    return result[0]['translation_text']

demo = gr.Interface(
    fn=translate_text,
    inputs=["text", "text", "text"],
    outputs="text",
    title="academic translator",
    description="translates from academic german to English"
)

demo.launch(share=True, mcp_server=True)