Spaces:
Sleeping
Sleeping
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) |