import spaces import gradio as gr from transformers import AutoModelForCausalLM, AutoTokenizer import torch from gradio_rich_textbox import RichTextbox title = """# Welcome to 🌟Tonic's🧬📏💪🏻Genstruct 7B ! 🧬📏💪🏻[Genstruct 7B](https://huggingface.co/NousResearch/Genstruct-7B) is an instruction-generation model, designed to create valid instructions given a raw text corpus. This enables the creation of new, partially synthetic instruction finetuning datasets from any raw-text corpus. You can build with this endpoint using🧬📏💪🏻[Genstruct 7B](https://huggingface.co/NousResearch/Genstruct-7B) available here : [NousResearch/Genstruct-7B](https://huggingface.co/NousResearch/Genstruct-7B). You can also use ✨StarCoder by cloning this space. Simply click here: Duplicate Space Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder's 🛠️community 👻 [![Join us on Discord](https://img.shields.io/discord/1109943800132010065?label=Discord&logo=discord&style=flat-square)](https://discord.gg/GWpVpekp) On 🤗Huggingface:[MultiTransformer](https://huggingface.co/MultiTransformer) Math 🔍 [introspector](https://huggingface.co/introspector) On 🌐Github: [Tonic-AI](https://github.com/tonic-ai) & contribute to🌟 [SciTonic](https://github.com/Tonic-AI/multitonic)🤗Big thanks to Yuvi Sharma and all the folks at huggingface for the community grant 🤗 """ examplecofee = "A cortado is a Spanish beverage consisting of espresso mixed with a roughly equal amount of warm milk to reduce the acidity,[1][2] although the exact ratios have considerable regional variation.[3] The milk in a cortado is steamed, but not frothy and "texturized" as in many Italian coffee drinks.[4] The cortado is commonly served all over Spain.[5] The word cortado is the past participle of the Spanish verb cortar (to cut), in the sense of "dilute", and can refer variously to either coffee or espresso drinks throughout Spanish and Portuguese speaking countries." model_path = "NousResearch/Genstruct-7B" # hf_token = os.getenv("HF_TOKEN") # if not hf_token: # raise ValueError("Hugging Face token not found. Please set the HF_TOKEN environment variable.") tokenizer = AutoTokenizer.from_pretrained(model_path) # quantization_config = BitsAndBytesConfig(load_in_8bit=True) model = AutoModelForCausalLM.from_pretrained( model_path, device_map='cuda', load_in_8bit=True) @spaces.GPU def generate_text(prompt, custom_prompt, temperature, max_length): # Example message format msg = [{ 'title': prompt, 'content': custom_prompt }] prompt_text = f"title: {msg[0]['title']}\ncontent: {msg[0]['content']}" inputs = tokenizer(prompt_text, return_tensors='pt').to('cuda') generated_ids = model.generate(**inputs, max_new_tokens=max_length, temperature=temperature, pad_token_id=tokenizer.eos_token_id) generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True) return generated_text def gradio_app(): with gr.Blocks() as demo: gr.Markdown(title) prompt = gr.Textbox(label="WordPhrases", value=examplecofee, lines=5) custom_prompt = gr.Textbox(label="Title", value="Cortado", lines=1) with gr.Row(): temperature = gr.Slider(minimum=0.1, maximum=1.0, step=0.1, value=0.5, label="Temperature") max_length = gr.Slider(minimum=250, maximum=1024, step=10, value=450, label="Generate Length") generate_btn = gr.Button("Try 🧬📏💪🏻 Genstruct") output = gr.RichTextbox(label="🧬📏💪🏻Genstruct 7B:", lines=20) generate_btn.click( fn=generate_text, inputs=[prompt, custom_prompt, temperature, max_length], outputs=output ) demo.launch() if __name__ == "__main__": gradio_app()