Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import nest_asyncio
 nest_asyncio.apply()
 import streamlit as st
-from transformers import pipeline, AutoTokenizer
+from transformers import pipeline
 from huggingface_hub import login
 from streamlit.components.v1 import html
 import pandas as pd
@@ -294,7 +294,7 @@ def main():
                 st.error(f"Gemma processing failed: {str(error)}")
                 report = "Error generating report. Please try again with fewer tweets."
             else:
-                report = raw_result[0]['generated_text']
+                report = extract_assistant_response(raw_result)
                 # report = extract_assistant_response(raw_result)
             clear_gpu_memory()
             progress_bar.progress(100)