indentention

metadata_transformer.py

@@ -14,16 +14,17 @@ You are a metadata schema translator. You translate metadata from one schema to
 """
 
 def translate(schema_input, schema_target):
-    propmpt = '"""{} \n Translate the schema metadata file above to the schema: {}"""'.format(schema_input, schema_target)    
-    response = huggingface.ChatCompletion.create(
-        model="mistralai/Mistral-7B-v0.1",
-        messages=[
-            {"role": "system", "content": system_message},
-            {"role": "user", "content": propmpt},
-        ],
-          temperature=0.9,
-          top_p=0.6,
-          max_tokens=256,
+    prompt = '"""{} \n Translate the schema metadata file above to the schema: {}"""'.format(schema_input, schema_target)    
+    response = huggingface.Completion.create(
+        model="meta-llama/Llama-2-7b-hf",
+        # messages=[
+        #     {"role": "system", "content": system_message},
+        #     {"role": "user", "content": prompt},
+        # ],
+        prompt=prompt,
+        temperature=0.9,
+        top_p=0.6,
+        max_tokens=256,
     )
     print(response)
     return response['choices'][0]['message']['content']