Rooni commited on
Commit
6a72c67
·
verified ·
1 Parent(s): 18777c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -28,15 +28,16 @@ def generate_cached_image(prompt, width, height, seed, model_name):
28
  "Content-Type": "application/json"
29
  }
30
  data = {
31
- "prompt": prompt,
32
- "width": width,
33
- "height": height,
34
- "seed": seed,
35
- "model": model_name
 
36
  }
37
 
38
  try:
39
- response = requests.post("https://api.your-provider.com/generate", headers=headers, json=data)
40
  response.raise_for_status()
41
  image_data = response.content
42
  image = Image.open(io.BytesIO(image_data))
@@ -49,8 +50,8 @@ def generate_cached_image(prompt, width, height, seed, model_name):
49
  def get_image(prompt):
50
  width = request.args.get('width', type=int, default=512)
51
  height = request.args.get('height', type=int, default=512)
52
- seed = request.args.get('seed', type=int, required=True)
53
- model_name = request.args.get('model', required=True)
54
 
55
  # Декодируем URL-кодированный prompt
56
  prompt = urllib.parse.unquote(prompt)
 
28
  "Content-Type": "application/json"
29
  }
30
  data = {
31
+ "inputs": prompt,
32
+ "parameters": {
33
+ "width": width,
34
+ "height": height,
35
+ "seed": seed
36
+ }
37
  }
38
 
39
  try:
40
+ response = requests.post(f"https://api-inference.huggingface.co/models/{model_name}", headers=headers, json=data)
41
  response.raise_for_status()
42
  image_data = response.content
43
  image = Image.open(io.BytesIO(image_data))
 
50
  def get_image(prompt):
51
  width = request.args.get('width', type=int, default=512)
52
  height = request.args.get('height', type=int, default=512)
53
+ seed = request.args.get('seed', type=int, default=22)
54
+ model_name = request.args.get('model', default="Shakker-Labs_FLUX.1-dev-LoRA-Logo-Design").replace('_', '/')
55
 
56
  # Декодируем URL-кодированный prompt
57
  prompt = urllib.parse.unquote(prompt)