Update app.py

app.py

@@ -4,12 +4,13 @@ import torch
 
 # تحميل النموذج والـ Tokenizer
 model_path = "./saved_model"
+tokenizer_path = "./saved_tokenizer"
 
 try:
+    tokenizer = T5Tokenizer.from_pretrained(tokenizer_path)
     model = T5ForConditionalGeneration.from_pretrained(model_path)
-    tokenizer = T5Tokenizer.from_pretrained(model_path)
-    device = torch.device("cpu")
-    model.to(device)
+    device = torch.device("cpu")  # تحديد الجهاز على CPU
+    model.to(device)  # نقل النموذج إلى CPU
     model_loaded = True
 except Exception as e:
     st.error(f"Error loading model: {e}")
@@ -19,64 +20,28 @@ except Exception as e:
 def generate_summary(text):
     try:
         inputs = ["summarize: " + text]
-        inputs = tokenizer(inputs, max_length=1024, truncation=True, return_tensors="pt").to(device)
+        inputs = tokenizer(inputs, max_length=1024, truncation=True, return_tensors="pt").to(device)  # نقل ال inputs الي ال CPU
         outputs = model.generate(inputs.input_ids, max_length=150, length_penalty=2.0, num_beams=4, early_stopping=True)
         return tokenizer.decode(outputs[0], skip_special_tokens=True)
     except Exception as e:
         st.error(f"Error generating summary: {e}")
         return None
 
-# تخصيص الألوان
-primary_color = "#FFB6C1"  # وردي فاتح
-secondary_color = "#E6E6FA"  # بنفسجي فاتح
-text_color = "#4A4A4A"  # رمادي داكن
+# واجهة المستخدم باستخدام Streamlit
+st.title("application summarize")
 
-# تخصيص واجهة المستخدم
-st.set_page_config(page_title="Text Summarizer", page_icon=":sparkles:", layout="wide")
+text = st.text_area("enter text you want to summarize.")
 
-st.markdown(
-    f"""
-    <style>
-        body {{
-            color: {text_color};
-            background-color: {secondary_color};
-        }}
-        .stApp {{
-            background-color: {secondary_color};
-        }}
-        .stButton button {{
-            background-color: {primary_color};
-            color: white;
-        }}
-        .stTextArea textarea {{
-            background-color: white;
-        }}
-        .stTextInput input {{
-            background-color: white;
-        }}
-    </style>
-    """,
-    unsafe_allow_html=True,
-)
-
-st.title("✨ Text Summarization Application ✨")
-
-st.markdown("Enter the text you want to summarize, and the application will generate a concise summary.")
-
-text = st.text_area("Enter text here:", height=200)
-
-if st.button("Summarize"):
+if st.button("summary:"):
     if text and model_loaded:
         with st.spinner("Generating summary..."):
             summary = generate_summary(text)
         if summary:
-            st.subheader("Summary:")
+            st.write("summary")
             st.write(summary)
         else:
             st.warning("Summary generation failed. Please check the input text.")
     elif not model_loaded:
         st.error("Model failed to load. Please check the application logs.")
     else:
-        st.warning("Please enter text.")
-
-st.markdown("---")
+        st.warning("Please enter text.")