up

app.py

@@ -9,14 +9,14 @@ model_pipelines = {}
 
 # Load a default model initially
 default_model_name = "gia-project/gia2-small-untrained"
-default_generator = pipeline("text-generation", model=default_model_name, tokenizer="gpt2", trust_remote_code=True)
+default_generator = pipeline("text-generation", model=default_model_name, trust_remote_code=True)
 model_pipelines[default_model_name] = default_generator
 
 def generate_text(model_name, input_text):
     # Check if the selected model is already loaded
     if model_name not in model_pipelines:
         # Load the model and create a pipeline if it's not already loaded
-        generator = pipeline("text-generation", model=model_name, tokenizer="gpt2", trust_remote_code=True)
+        generator = pipeline("text-generation", model=model_name, trust_remote_code=True)
         model_pipelines[model_name] = generator
     
     # Get the pipeline for the selected model and generate text
@@ -28,7 +28,7 @@ def generate_text(model_name, input_text):
 iface = gr.Interface(
     fn=generate_text,  # Function to be called on user input
     inputs=[
-        gr.inputs.Dropdown(choices=models, label="Select Model"),  # Dropdown to select model
+        gr.inputs.Dropdown(choices=models, label="Select Model", default=default_model_name),  # Dropdown to select model
         gr.inputs.Textbox(lines=5, label="Input Text")  # Textbox for entering text
     ],
     outputs=gr.outputs.Textbox(label="Generated Text"),  # Textbox to display the generated text