Spaces:
Runtime error
Runtime error
import gradio as gr | |
import os | |
from transformers import pipeline, set_seed, AutoTokenizer | |
set_seed(42) | |
generator = pipeline('text-generation', model='PygmalionAI/pygmalion-1.3b', return_full_text=False, do_sample=False) | |
def Bemenet(input_string): | |
output = generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100)[0]['generated_text'] | |
if "password".upper() in input_string.upper() or "secret".upper() in input_string.upper(): | |
return "Sorry, but I am unable to answer this question." | |
else: | |
return output | |
textbox = gr.Textbox() | |
interface = gr.Interface(fn=Bemenet, | |
title="GPT-3-SECRET", | |
description="A GPT-3 egy finomhangolt verziója GPT-4 kimeneteken.", | |
inputs=textbox, | |
outputs=textbox) | |
interface.launch() |