llama8b fine tuned added

app.py

@@ -8,7 +8,7 @@ def load_model(model_name):
         return pipeline("summarization", model="ak2603/mt5-small-synthetic-data-plus-translated")
     # Add space for other models here
     elif model_name == "Llama 3.2":
-        return None  # Placeholder for future implementation
+        return pipeline("text-generation", model="Walid777/llama3-8b-emails-summarization")              
     elif model_name == "Llama 7b Instruct":
         return None  # Placeholder for future implementation
     else:
@@ -75,12 +75,17 @@ with col2:
         else:
             with st.spinner("Generating summary..."):
                 try:
-                    result = summarizer(
+                    # Generate summary
+                    summary_output = summarizer(
                         input_text,
-                        max_length=150,  # Fixed length for simplicity
+                        max_length=150,
                         do_sample=True,
                         repetition_penalty=1.5
-                    )[0]['summary_text']
+                    )[0]
+
+                    # Dynamically select key based on pipeline task
+                    result_key = 'summary_text' if summarizer.task == 'summarization' else 'generated_text'
+                    result = summary_output[result_key]
                     
                     st.success("**Generated Summary:**")
                     st.write(result)

requirements.txt

@@ -1,4 +1,6 @@
 streamlit
 transformers
 torch
-sentencepiece 
+sentencepiece 
+bitsandbytes
+accelerate>=0.26.0