eaglelandsonce commited on
Commit
7bf061b
1 Parent(s): 8793735

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -47,7 +47,6 @@ GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_API_KEY')
47
 
48
  # Function to generate image using Clarifai
49
  def generate_image(prompt):
50
- size = (512,512)
51
  channel = ClarifaiChannel.get_grpc_channel()
52
  stub = service_pb2_grpc.V2Stub(channel)
53
  metadata = (('authorization', 'Key ' + PAT),)
@@ -58,7 +57,7 @@ def generate_image(prompt):
58
  user_app_id=userDataObject,
59
  model_id=MODEL_ID_IMG,
60
  version_id=MODEL_VERSION_ID_IMG,
61
- inputs=[resources_pb2.Input(data=resources_pb2.Data(text=resources_pb2.Text(raw=prompt), image=resources_pb2.Image(size=size)))]
62
  ),
63
  metadata=metadata
64
  )
 
47
 
48
  # Function to generate image using Clarifai
49
  def generate_image(prompt):
 
50
  channel = ClarifaiChannel.get_grpc_channel()
51
  stub = service_pb2_grpc.V2Stub(channel)
52
  metadata = (('authorization', 'Key ' + PAT),)
 
57
  user_app_id=userDataObject,
58
  model_id=MODEL_ID_IMG,
59
  version_id=MODEL_VERSION_ID_IMG,
60
+ inputs=[resources_pb2.Input(data=resources_pb2.Data(text=resources_pb2.Text(raw=prompt)))]
61
  ),
62
  metadata=metadata
63
  )