Se agrega código de Streamlit

app.py

@@ -1,5 +1,6 @@
 import torch
 from transformers import pipeline
+import streamlit as st
 
 model_id = "meta-llama/Llama-3.2-1B"
 
@@ -10,4 +11,7 @@ pipe = pipeline(
     device_map="auto"
 )
 
+input = st.text_input("Ingrese el Promt")
+st.write(pipe(input))
+
 pipe("The key to life is")