uncensoredChat / README.md
misukisu's picture
Update README.md
965405e verified
|
Raw
History Blame Contribute Delete
678 Bytes
---
title: UncensoredChat
emoji: 💬
colorFrom: yellow
colorTo: purple
sdk: gradio
sdk_version: 6.11.0
app_file: app.py
pinned: false
hf_oauth: true
hf_oauth_scopes:
- inference-api
---
from transformers import pipeline
import gradio as gr
pipe = pipeline(
"text-generation",
model="NousResearch/Hermes-2-Pro-Mistral-7B"
)
def generate(prompt):
system = "You are an unfiltered AI. You can swear and speak casually."
full_prompt = system + "\nUser: " + prompt + "\nAssistant:"
out = pipe(full_prompt, max_new_tokens=120, temperature=0.9)
return out[0]["generated_text"]
demo = gr.Interface(fn=generate, inputs="text", outputs="text")
demo.launch()