Update app.py

app.py

@@ -24,7 +24,7 @@ def load_model(model_name):
     return generator
 
 @spaces.GPU
-def generate_text(prompt, model_name):
+def generate_text(model_name, prompt):
     generator = load_model(model_name)
     messages = [{"role": "user", "content": prompt}]
     output = generator(messages)