Create app.py

Initial version

app.py

@@ -0,0 +1,17 @@
+import streamlit as st
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# Load the model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
+
+st.title("Llama 3.2 3B Instruct Text Generation")
+st.write("Enter a prompt and generate text using theLlama 3.2 3B model.")
+
+user_input = st.text_area("Enter your prompt:", "Here is your input...")
+
+if st.button("Generate"):
+    inputs = tokenizer(user_input, return_tensors="pt")
+    outputs = model.generate(inputs["input_ids"], max_length=200)
+    generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    st.write(generated_text)