Spaces:
Runtime error
Runtime error
import gradio as gr | |
from langchain.llms import OpenAI | |
from langchain.prompts import PromptTemplate | |
from langchain.chains.llm import LLMChain | |
from langchain.chains.constitutional_ai.base import ConstitutionalChain | |
from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple | |
def yodafy(sentence, selection): | |
llm = OpenAI(model_name="gpt-3.5-turbo", temperature=.8) | |
prompt = PromptTemplate( | |
input_variables=["sentence", "selection"], | |
template="You are Master Yoda. A young apprentice has came to tell ysou this sentence: {sentence}. {selection} the sentence as Master Yoda would. You may refer to the Star Wars and use punctuation. Remove any \n.", | |
) | |
chain = LLMChain(llm=llm, prompt=prompt) | |
master_yoda_principle = ConstitutionalPrinciple( | |
name='Master Yoda Principle', | |
critique_request='Identify specific ways in which the model\'s response is not in the style of Master Yoda.', | |
revision_request='Please rewrite the model response to be in the style of Master Yoda using his teachings, his wisdom and the Force.', | |
) | |
constitutional_chain = ConstitutionalChain.from_llm( | |
chain=chain, | |
constitutional_principles=[master_yoda_principle], | |
llm=llm, | |
verbose=True, | |
) | |
return constitutional_chain.run(sentence=sentence, selection=selection) | |
iface = gr.Interface(fn=yodafy, | |
inputs=["text", | |
gr.inputs.Radio(["Rewrite", "Reply"])], | |
outputs="text", | |
examples=[ | |
["May the Force be with you!", "Rewrite"], | |
] | |
) | |
iface.launch() |