Spaces:

ToletiSri
/

TSAI_S27

Sleeping

ToletiSri commited on Dec 21, 2023

Commit

6da9f6d

•

1 Parent(s): ba7f947

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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)
 tokenizer.pad_token = tokenizer.eos_token
 def inference(prompt, count):

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+model = AutoModelForCausalLM.from_pretrained("checkpoint_500",trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("checkpoint_500", trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token
 def inference(prompt, count):