Update app.py

app.py

@@ -6,8 +6,8 @@ class RinnaTalk():
     def __init__(self, tokenizer=None, model=None):
         self.prompt = ''
         # 事前にmodelとtokenizerを呼んでおく
-        self.tokenizer = AutoTokenizer.from_pretrained("Hoshiyume/FixedStar-BETA", use_fast=False) if tokenizer is None else tokenizer
-        self.model = AutoModelForCausalLM.from_pretrained("Hoshiyume/FixedStar-BETA", torch_dtype=torch.float16) if model is None else model
+        self.tokenizer = AutoTokenizer.from_pretrained("models", use_fast=False) if tokenizer is None else tokenizer
+        self.model = AutoModelForCausalLM.from_pretrained("models", torch_dtype=torch.float16) if model is None else model
 
     def chat(self, message: str, chat_history: list, max_token_length: int = 128, min_token_length: int = 10, temperature: float = 0.8):
         # チャット履歴をクリアした際にpromptもクリアさせるため