chywang commited on
Commit
f1f9eb4
1 Parent(s): f9c3d7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,7 +70,7 @@ def easyphoto_train(instance_images):
70
  images.append(image_url)
71
  client = ApiClient(host, appId, token)
72
  api = AIGCImagesApi(client)
73
- response = api.aigc_images_train(images, '', None)
74
  message = response.message
75
  model_id = response.data['model_id']
76
  job_id = response.data['job_id']
@@ -141,7 +141,7 @@ def easyphoto_infer(model_id, selected_template_images, additional_prompt, seed,
141
  "color_shift_last": color_shift_last,
142
  "background_restore": background_restore
143
  }
144
- response = api.aigc_images_create(model_id, image_url, 'photog_infer_with_webui_pmml', params)
145
  except:
146
  output_info = 'Status: calling eas service failed!'
147
  return output_info, []
 
70
  images.append(image_url)
71
  client = ApiClient(host, appId, token)
72
  api = AIGCImagesApi(client)
73
+ response = api.aigc_images_train(images, 'photog_train_freetier', None)
74
  message = response.message
75
  model_id = response.data['model_id']
76
  job_id = response.data['job_id']
 
141
  "color_shift_last": color_shift_last,
142
  "background_restore": background_restore
143
  }
144
+ response = api.aigc_images_create(model_id, image_url, 'photog_infer_freetier', params)
145
  except:
146
  output_info = 'Status: calling eas service failed!'
147
  return output_info, []