Update app.py

app.py

@@ -1,28 +1,38 @@
-import gradio as gr
-import tensorflow as tf
-from SpaceGen_preprocessing import *
-from utils import *
-
-
-# Path to your Keras model
-model_path = "SpaceGen_Large.keras"
-
-# Load the model
-model = tf.keras.models.load_model(model_path)
-
-
-def fix_space(text):
-  text = clean_sentence(text)
-  X = text_to_X(text)
-  predictions = model.predict(X, verbose=0)
-  predicted_labels = []
-  for pred in predictions[0]:
-    predicted_labels.append(1 if pred[1] > .5 else 0)
-  fixed_text = insert_spaces(text.replace(' ',''), find_indices(predicted_labels))
-  return fixed_text
-
-default_text = "T hel ittlegi rlra nthro ughth epa rkc has ing abut terfly."
-demo = gr.Interface(fn=fix_space,
-                    inputs=gr.Textbox(label="Input Text", value=default_text),
-                    outputs="text")
-demo.launch()
+import gradio as gr
+import tensorflow as tf
+from SpaceGen_preprocessing import *
+from utils import *
+
+# טוען את המודל
+model_path = "SpaceGen_Large.keras"
+model = tf.keras.models.load_model(model_path)
+
+# פונקציית תיקון רווחים
+def fix_space(text):
+    text = clean_sentence(text)
+    X = text_to_X(text)
+    predictions = model.predict(X, verbose=0)
+    predicted_labels = []
+    for pred in predictions[0]:
+        predicted_labels.append(1 if pred[1] > .5 else 0)
+    fixed_text = insert_spaces(text.replace(' ',''), find_indices(predicted_labels))
+    return fixed_text
+
+default_text = "T hel ittlegi rlra nthro ughth epa rkc has ing abut terfly."
+
+description = (
+    "פרויקט לימודי במדעי הנתונים ומדעי המוח.\n"
+    "המערכת מדגימה שימוש ברשת LSTM לזיהוי מיקום רווחים חסרים בטקסט.\n"
+    "נכון לעכשיו, המודל תומך באנגלית בלבד.\n"
+    "לשאלות וביקורות: [email protected]"
+)
+
+demo = gr.Interface(
+    fn=fix_space,
+    inputs=gr.Textbox(label="Input Text", value=default_text),
+    outputs="text",
+    title="SpaceGen – Text Space Correction with LSTM",
+    description=description
+)
+
+demo.launch()