Update app

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 # Define the function to handle text generation
-def generate_text(model_name, text, num_beams, max_length, top_p, temperature, repetition_penalty, no_repeat_ngram_size):
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+def generate_text(model_name, text, num_beams, max_length, top_p, temperature, repetition_penalty, no_repeat_ngram_size, token):
+    tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
+    model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token)
     
     # Initialize pipeline with explicit model and tokenizer
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
@@ -32,7 +32,8 @@ inputs_component = [
     gr.Slider(minimum=0.1, maximum=1.0, step=0.1, label="Top p"),
     gr.Slider(minimum=0.1, maximum=1.0, step=0.1, label="Temperature"),
     gr.Slider(minimum=1.0, maximum=5.0, step=0.5, label="Repetition Penalty"),
-    gr.Slider(minimum=2, maximum=5, step=1, label="No Repeat Ngram Size")
+    gr.Slider(minimum=2, maximum=5, step=1, label="No Repeat Ngram Size"),
+    gr.Textbox(placeholder="Enter your Hugging Face token here...", label="Hugging Face Token", type="password")
 ]
 
 # Setup the interface with live=False to require button press