Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-model = AutoModelForCausalLM.from_pretrained("checkpoint",trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained("checkpoint", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("checkpoint/",trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("checkpoint/", trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token
 
 def inference(prompt, count):