Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 from transformers import pipeline
 
+# Ensure the correct Keras package is used
+import tensorflow as tf
+import tf_keras
+
 # Define the model and the text generation function
 fine_tuned_model = "Amitesh007/text_generation-finetuned-gpt2"
 generator = pipeline('text-generation', model=fine_tuned_model)
@@ -29,8 +33,8 @@ with gr.Blocks() as demo:
         inputs=input_text
     )
     
-    generate_button.click(fn=generate, inputs=input_text, outputs=[output_text1, output_text2])
+    generate_button.click(fn=generate, inputs=input_text, outputs=[output_text1, output_text2], concurrency_limit=2)
+
+# Launch the interface with the max_threads parameter to control the total number of workers
+demo.launch(max_threads=8)
 
-# Launch the interface
-demo.queue(concurrency_count=2)
-demo.launch()