Update app.py

app.py

@@ -1,17 +1,16 @@
+import os
 import gradio as gr
+from huggingface_hub import login
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Load LLaMA 3.2 model and tokenizer
-model_name = "meta-llama/Llama-3.2-1B"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
-
-# Fetch API token from environment
+# Fetch API token from environment variable
 api_token = os.getenv("Llama_Token")
 
-# Authenticate
+# Authenticate with Hugging Face
 login(api_token)
 
+# Load LLaMA 3.2 model and tokenizer with the API token
+model_name = "meta-llama/Llama-3.2-1B"
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=api_token)
 model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=api_token)
 
@@ -25,9 +24,9 @@ def generate_text(prompt, max_length=100, temperature=0.7):
 iface = gr.Interface(
     fn=generate_text,
     inputs=[
-        gr.inputs.Textbox(label="Enter your prompt", placeholder="Start typing..."),
-        gr.inputs.Slider(minimum=50, maximum=200, default=100, label="Max Length"),
-        gr.inputs.Slider(minimum=0.1, maximum=1.0, default=0.7, label="Temperature"),
+        gr.Textbox(label="Enter your prompt", placeholder="Start typing..."),
+        gr.Slider(minimum=50, maximum=200, default=100, label="Max Length"),
+        gr.Slider(minimum=0.1, maximum=1.0, default=0.7, label="Temperature"),
     ],
     outputs="text",
     title="LLaMA 3.2 Text Generator",