izuemon commited on
Commit
252f690
·
verified ·
1 Parent(s): 66807f1

Update turbowarp-server/qr-converter.py

Browse files
Files changed (1) hide show
  1. turbowarp-server/qr-converter.py +7 -17
turbowarp-server/qr-converter.py CHANGED
@@ -101,23 +101,13 @@ def rgb_to_scratch_number(rgb):
101
  return f"{r_scaled:02d}{g_scaled:02d}{b_scaled:02d}"
102
 
103
  def generate_image(prompt):
104
- """APIから352x352の画像生成"""
105
- seed = "10900"
106
- print(f"[INFO] Generating image with prompt: '{prompt[:50]}...', seed={seed}")
107
- params = {
108
- "prompt": prompt,
109
- "negative_prompt": "nsfw, low quality",
110
- "width": 352,
111
- "height": 352,
112
- "num_inference_steps": 2,
113
- "guidance_scale": 0,
114
- "seed": seed,
115
- "randomize_seed": "true"
116
- }
117
- resp = requests.get("https://izuemon-pixart-alpha-pixart-sigma-xl-2-1024-ms.hf.space/gen", params=params)
118
- resp.raise_for_status()
119
- img = Image.open(BytesIO(resp.content))
120
- print(f"[INFO] Image generated: size={img.size}, mode={img.mode}")
121
  return img
122
 
123
  def resize_and_encode(img):
 
101
  return f"{r_scaled:02d}{g_scaled:02d}{b_scaled:02d}"
102
 
103
  def generate_image(prompt):
104
+ """テスト用:ローカル画像を読み込む"""
105
+ path = "turbowarp-server/test_result.png"
106
+ print(f"[TEST] Loading test image from '{path}' (prompt was: '{prompt[:50]}...')")
107
+
108
+ img = Image.open(path)
109
+ print(f"[INFO] Test image loaded: size={img.size}, mode={img.mode}")
110
+
 
 
 
 
 
 
 
 
 
 
111
  return img
112
 
113
  def resize_and_encode(img):