update app

app.py

@@ -1,4 +1,6 @@
 from transformers import BartForConditionalGeneration, BartTokenizer
+
+
 import gradio as gr
 
 # Charger le modèle BART et le tokenizer
@@ -12,7 +14,13 @@ def generate_text(prompt):
     summary_ids = model.generate(inputs["input_ids"], max_length=150, min_length=40, length_penalty=2.0, num_beams=4, early_stopping=True)
     return tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
+#for training the model after the data is collected
+#model.save_pretrained("model")
+#tokenizer.save_pretrained("model")
+
 # Créer une interface de saisie avec Gradio
 interface = gr.Interface(fn=generate_text, inputs="text", outputs="text",title="TeLLMyStory",description="Enter your story idea and the model will generate the story based on it.")
+
+
 #Lancer l'interface
-interface.launch()
+interface.launch(share=True)