Update app.py

app.py

@@ -1,8 +1,25 @@
 import gradio as gr
-import os
+from transformers import pipeline
 
-hf_token = os.environ.get('HF_TOKEN')
-gr.Interface.load("models/Karzan/bart-base-quran-transliteration",
-                  hf_token=hf_token,
-                 inputs=gr.Textbox(lines=5, label="Input Text")  # customizes the input component
-                 ).launch()
+# model = AutoModelForCausalLM.from_pretrained("Karzan/ckb-gpt2-medium-base-test-1024")
+model_id = "Karzan/bart-base-quran-transliteration"
+pipe = pipeline("text2text-generation", model=model_id, max_length=1024)
+
+
+def casual_model_function(input_text):
+    # Replace the line below with the code for your casual model
+    output_text = pipe(input_text)[0]['generated_text']
+    return output_text
+text_input = gr.Textbox(lines=10, placeholder="Enter your text here", label="Your Text")
+text_outputs = gr.Textbox(lines=10)
+# submit_btn = gr.Button(value="Generate",variant="secondary")
+
+interface = gr.Interface(
+    fn=casual_model_function,
+    inputs=text_input,
+    outputs=text_outputs,
+    # submit_btn=submit_btn,
+)
+
+if __name__ == "__main__":
+    interface.launch()