Update app.py

app.py

@@ -1,10 +1,20 @@
 from transformers import pipeline
 import gradio as gr
 
-messages = [
-    {"role": "user", "content": "Who are you?"},
-]
+pipeline = pipeline(model = "meta-llama/Llama-3.2-1B-Instruct")
 
-pipe = pipeline(model="meta-llama/Llama-3.2-1B-Instruct")
+def inference(input):
+    output = pipeline(input)
+    return output
 
-pipe(messages, max_length=512)
+app = gr.Interface(
+    fn = inference,
+    inputs = [gr.Textbox(label = "Input")],
+    outputs = [gr.Textbox(label = "Output")],
+    title = "Demo",
+    examples = [
+        ["Hello, World."]
+    ]
+)
+
+app.launch()