Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 
 # Initialize the text generation pipeline
-generator = pipeline("text-generation", model="gpt2", tokenizer="gpt2")
+generator = pipeline("text-generation", model="isitcoding/gpt2_120_finetuned", tokenizer="isitcoding/gpt2_120_finetuned")
 
 # Define the response function with additional options for customization
 def text_generation(
@@ -20,7 +20,7 @@ def text_generation(
 ):
     # Setup the configuration for the model generation
     gen_params = {
-        "max_length": 50,  # Default, you can tweak it or set from parameters
+        "max_length": 518,  # Default, you can tweak it or set from parameters
         "num_return_sequences": 1,
         "do_sample": do_sample,
         "temperature": 0.7,  # Controls randomness