Update app.py
Browse files
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)
|
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 |
)
|