FuturesonyAi / app.py
Futuresony's picture
Update app.py
9d65752 verified
raw
history blame
927 Bytes
import gradio as gr
from huggingface_hub import InferenceClient
client = InferenceClient("Futuresony/future_ai_12_10_2024.gguf")
def respond(message, history, max_tokens, temperature, top_p):
"""Sends user input directly to the model without extra formatting."""
response = client.text_generation(
message.strip(), # βœ… Sends only the question
max_new_tokens=max_tokens,
temperature=temperature,
top_p=top_p,
)
yield response.strip() # βœ… Removes unnecessary spaces/newlines
demo = gr.ChatInterface(
respond,
additional_inputs=[
gr.Slider(minimum=1, maximum=250, value=128, step=1, label="Max new tokens"),
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
],
)
if __name__ == "__main__":
demo.launch()