Update app.py

app.py

@@ -1,12 +1,21 @@
 import gradio as gr
 from transformers import pipeline
 
-gr.load("models/codellama/CodeLlama-7b-hf").launch()
+# Load the text generation pipeline with the CodeLlama model
+text_generation_pipeline = pipeline("text-generation", model="codellama/CodeLlama-70b-Instruct-hf")
 
-# Prompt the user for input
-user_input = input("AI: Ask me anything\nMe: ")
+# Define a function to generate responses based on user input
+def generate_response(input_text):
+    # Generate a response using the pipeline
+    generated_response = text_generation_pipeline(input_text, max_length=200)[0]['generated_text']
+    return generated_response
 
-# Generate a response using the pipeline
-generated_response = text_generation_pipeline(user_input, max_length=200)[0]['generated_text']
+# Create Gradio interface
+gr.Interface(
+    fn=generate_response,
+    inputs="text",
+    outputs="text",
+    title="CodeLlama Assistant",
+    description="Ask me anything and I will respond!",
+).launch()
 
-print("AI:", generated_response)