variables

Author: royjhan

examples/fill-in-middle/main.py

@@ -1,9 +1,16 @@
 from ollama import generate
 
+prompt = '''def remove_non_ascii(s: str) -> str:
+    """ '''
+
+suffix = """
+    return result
+"""
+
 response = generate(
   model='codellama:7b-code',
-  prompt='def remove_non_ascii(s: str) -> str:',
-  suffix='return result',
+  prompt=prompt,
+  suffix=suffix,
   options={
     'num_predict': 128,
     'temperature': 0,