Update app.py

app.py

@@ -3,9 +3,10 @@ import os
 os.system("pip install transformers sentencepiece torch")
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("yniw/open-calm-7b-4gib")
-model = AutoModelForCausalLM.from_pretrained("yniw/open-calm-7b-4gib")
 
+tokenizer = AutoTokenizer.from_pretrained("kyo-takano/open-calm-7b-8bit")
+
+model = AutoModelForCausalLM.from_pretrained("kyo-takano/open-calm-7b-8bit")
 def generate_text(input_text, temperature=0.8, max_length=20):
     input_ids = tokenizer.encode(input_text, return_tensors="pt")
     output = model.generate(input_ids, max_length=max_length, temperature=temperature)