Update app.py

app.py

@@ -3,53 +3,7 @@ import os
 from huggingface_hub import login
 from transformers import pipeline
 
-# Login using your Hugging Face API key
 api_key = os.getenv("LLAMA")
 login(api_key)
 
-# Load the model using the Hugging Face Hub
-llama_model = gr.load("models/meta-llama/Llama-3.1-8B-Instruct")
-
-# Function to handle input and output
-def chat_with_llama(user_input):
-    # Ensure user_input is correctly formatted as a string
-    if isinstance(user_input, str):
-        response = llama_model(user_input)
-        return response[0]  # Return the first element of the output list
-    else:
-        return "Invalid input. Please provide a valid text."
-
-# Customize the Gradio interface
-with gr.Blocks(css=".title {font-size: 3em; font-weight: bold; text-align: center; color: #4A90E2;}") as demo:
-    
-    # Header
-    gr.Markdown(
-        """
-        <div class="title">🦙 Llama 3.1 Chatbot 🦙</div>
-        <p style="text-align:center; font-size:1.2em; color:gray;">Ask anything from the advanced Llama 3.1 model!</p>
-        """,
-        elem_classes="header"
-    )
-    
-    # Main Input/Output Section
-    with gr.Row():
-        with gr.Column(scale=1):
-            user_input = gr.Textbox(label="Your question", placeholder="Type your question here...", lines=4)
-        with gr.Column(scale=1):
-            response_output = gr.Textbox(label="Llama's response", lines=4)
-
-    # Button to submit input
-    submit_button = gr.Button("Submit", variant="primary")
-
-    # Link the input and output
-    submit_button.click(fn=chat_with_llama, inputs=user_input, outputs=response_output)
-
-    # Footer
-    gr.Markdown(
-        """
-        <div style="text-align:center; font-size:0.8em; color:gray;">Developed with ❤️ using Llama 3.1 and Gradio</div>
-        """
-    )
-
-# Launch the app
-demo.launch()
+gr.load("models/meta-llama/Llama-3.1-8B-Instruct").launch()