simple complet

metadata_transformer.py

@@ -9,22 +9,18 @@ login(token=TOKEN)
 
 huggingface.prompt_builder = build_llama2_prompt
 
-system_message = """
-You are a metadata schema translator. You translate metadata from one schema to another.
-"""
+# system_message = """
+# You are a metadata schema translator. You translate metadata from one schema to another.
+# """
 
 def translate(schema_input, schema_target):
     prompt = '"""{} \n Translate the schema metadata file above to the schema: {}"""'.format(schema_input, schema_target)    
-    response = huggingface.ChatCompletion.create(
+    response = huggingface.Completion.create(
         model="meta-llama/Llama-2-7b-hf",
-        messages=[
-            {"role": "system", "content": system_message},
-            {"role": "user", "content": prompt},
-        ],
-        # prompt=prompt,
+        prompt=prompt,
         temperature=0.9,
         top_p=0.6,
         max_tokens=256,
     )
-    print(response)
+    
     return response['choices'][0]['message']['content']