Update demo.py

demo.py

@@ -77,7 +77,7 @@ locale.getpreferredencoding = lambda: "UTF-8"
 from chroma import Chroma, Protein, conditioners
 from chroma.models import graph_classifier, procap
 from chroma.utility.api import register_key
-from chroma.utility.chroma import letter_to_point_cloud, plane_split_protein
+from chroma_gen.chroma.utility.chroma import letter_to_point_cloud, plane_split_protein
 
 register_key(api_key)
 device = "cuda"