added code for llama

app.py

@@ -1,8 +1,22 @@
-import streamlit as st
+# import streamlit as st
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+# x = st.slider('Select a value')
+# st.write(x, 'squared is', x * x)
 
 # from datasets import load_dataset
 
-# dataset = load_dataset("soulhq-ai/insuranceQA-v2")
+# dataset = load_dataset("soulhq-ai/insuranceQA-v2")
+
+import torch
+from transformers import pipeline
+
+model_id = "meta-llama/Llama-3.2-3B"
+
+pipe = pipeline(
+    "text-generation", 
+    model=model_id, 
+    torch_dtype=torch.bfloat16, 
+    device_map="auto"
+)
+
+pipe("The key to life is")