Update app.py

app.py

@@ -89,8 +89,7 @@ def get_model_list():
     model_files = [f for f in model_files if f.endswith('.pt')]
     return model_files
 
-def gen(input, model_name):
-    model = load_model(model_name)
+def gen(input):
     generated_text = ''
     start_ids = encode(add_caseifer(input))
     x = (torch.tensor(start_ids, dtype=torch.long, device=device)[None, ...])
@@ -117,7 +116,7 @@ with gr.Blocks() as demo:
     def bot(history):
         # Generate the bot's response using your model
         input_text = history[-1][0]
-        bot_message = gen(input_text, model)
+        bot_message = gen(input_text)
         
         history[-1][1] = ""
         for character in bot_message: