Spaces:
Runtime error
Runtime error
# -*- coding: utf-8 -*- | |
"""Any Language Translator.ipynb | |
Automatically generated by Colaboratory. | |
Original file is located at | |
https://colab.research.google.com/drive/1cJmy7eHRvwkQJDWOg775L2iSMjfR4dFx | |
""" | |
import os | |
import openai | |
import gradio as gr | |
from dotenv import load_dotenv, find_dotenv | |
_ = load_dotenv(find_dotenv()) # read local .env file | |
os.environ["OPENAI_API_KEY"] = "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" | |
openai.api_key = os.environ['OPENAI_API_KEY'] | |
from langchain.chat_models import ChatOpenAI | |
from langchain.chains import LLMChain | |
from langchain.prompts import PromptTemplate | |
llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0) | |
translation_template = "Translate the following text from {source_language} to {target_language}: {text}" | |
translation_prompt = PromptTemplate(input_variables=["source_language", "target_language", "text"], template=translation_template) | |
translation_chain = LLMChain(llm=llm, prompt=translation_prompt) | |
def translate(source_language: str, target_language: str, text: str) -> str: | |
translated_text = translation_chain.predict(source_language=source_language, target_language=target_language, text=text) | |
return translated_text | |
inputs = [ | |
gr.inputs.Textbox(label="Source Language"), | |
gr.inputs.Textbox(label="Target Language"), | |
gr.inputs.Textbox(label="Text to Translate") | |
] | |
output = gr.outputs.Textbox(label="Translated Text") | |
gr.Interface(fn=translate, inputs=inputs, outputs=output).launch() |