simplfy

app.py

@@ -16,8 +16,7 @@ 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
-        model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
-        generator = pipeline("text-generation", model=model, tokenizer="gpt2", trust_remote_code=True)
+        generator = pipeline("text-generation", model=model_name, tokenizer="gpt2", trust_remote_code=True)
         model_pipelines[model_name] = generator
     
     # Get the pipeline for the selected model and generate text