KingNish commited on
Commit
9941ce1
1 Parent(s): dc60483

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,7 +72,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
72
  return seed
73
 
74
  @spaces.GPU(duration=30, queue=False)
75
- def king(type = "Image Editing",
76
  input_image = None,
77
  instruction: str = "Eiffel tower",
78
  steps: int = 8,
@@ -120,8 +120,8 @@ def king(type = "Image Editing",
120
 
121
  def response(instruction, input_image=None):
122
  if input_image is None:
123
- output=instruction
124
  output="Image Generation"
 
125
  else:
126
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
127
 
 
72
  return seed
73
 
74
  @spaces.GPU(duration=30, queue=False)
75
+ def king(type = "Image Generation",
76
  input_image = None,
77
  instruction: str = "Eiffel tower",
78
  steps: int = 8,
 
120
 
121
  def response(instruction, input_image=None):
122
  if input_image is None:
 
123
  output="Image Generation"
124
+ yield output
125
  else:
126
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
127