Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 os.system("python -m pip install --upgrade pip")
 os.system("pip install git+https://github.com/bigscience-workshop/petals")
 
-from petals import DistributedBloomForCausalLM
+from petals import AutoDistributedModelForCausalLM
 import npc_data
 
 # Choose any model available at https://health.petals.dev
@@ -15,7 +15,7 @@ model_name = "daekeun-ml/Llama-2-ko-instruct-13B"
 #quantumaikr/llama-2-70b-fb16-korean
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = DistributedBloomForCausalLM.from_pretrained(model_name)
+model = AutoDistributedModelForCausalLM.from_pretrained(model_name)
 
 # Run the model as if it were on your computer
 def chat2(id, npc, text):