uncensoredChat / README.md
misukisu's picture
Update README.md
965405e verified
|
Raw
History Blame Contribute Delete
678 Bytes

A newer version of the Gradio SDK is available: 6.25.0

Upgrade
metadata
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()