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 |
)
|