Spaces:

ejschwartz
/

aidapal-space

Running on Zero

ejschwartz commited on Apr 17

Commit

87253bd

1 Parent(s): 7a65a6f

Don't indent!

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,10 +36,10 @@ Analyze the following IDA Hex Rays pseudocode and generate a valid JSON object c
 @spaces.GPU
 def predict(code):
     prompt = f"""{system}
-    [INST]
-    {code}
-    [/INST]
-    """
     print(f"Prompt: {prompt}")
     print(f"Tokenized: {tokenizer.tokenize(prompt)}")

 @spaces.GPU
 def predict(code):
     prompt = f"""{system}
+[INST]
+{code}
+[/INST]
+"""
     print(f"Prompt: {prompt}")
     print(f"Tokenized: {tokenizer.tokenize(prompt)}")