Update sketch/gen_sketch.py

sketch/gen_sketch.py

@@ -66,7 +66,7 @@ class SketchApp:
         self.model = "gpt-4o"
         self.input_prompt = sketch_first_prompt.format(concept=args.concept_to_draw, gt_sketches_str=gt_example)
         self.gen_mode = args.gen_mode
-        self.seed_mode = args.seed_mode
+        self.temperature = args.temperature
         
 
     def call_llm(self, system_message, other_msg, additional_args):
@@ -74,7 +74,7 @@ class SketchApp:
             model=self.model,
             messages=[{"role": "system", "content": system_message}] + other_msg,
             max_tokens=self.max_tokens,
-            temperature=additional_args.get("temperature", 0.0),
+            temperature=self.temperature,
             stop=additional_args.get("stop", None)
         )
         return response.choices[0].message.content