Spaces:
Running
Running
File size: 1,665 Bytes
9fb9ed2 f072c32 9fb9ed2 f072c32 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import requests
import json
# API Endpoint
API_URL = "https://82le1lsl6e.execute-api.us-east-1.amazonaws.com/dev/test_bedrock_v5"
def invoke_text_api(prompt, max_tokens=500, temperature=0.7, top_k=100, top_p=0.95):
"""
Calls the API for text-only input.
"""
payload = {
"prompt": prompt,
"max_tokens": max_tokens,
"temperature": temperature,
"top_k": top_k,
"top_p": top_p
}
headers = {"Content-Type": "application/json"}
try:
response = requests.post(API_URL, json=payload, headers=headers)
response_json = response.json()
# Extract text response
return response_json.get("model_response", {}).get("content", [{}])[0].get("text", "🤖 No response received.")
except Exception as e:
return f"⚠️ Error: {str(e)}"
def invoke_text_image_api(base64_image, prompt, max_tokens=1000, temperature=0.8, top_k=150, top_p=0.98):
"""
Calls the API for text+image input.
"""
payload = {
"image": base64_image,
"media_type": "image/jpeg", # Ensure correct media type
"prompt": prompt,
"max_tokens": max_tokens,
"temperature": temperature,
"top_k": top_k,
"top_p": top_p
}
headers = {"Content-Type": "application/json"}
try:
response = requests.post(API_URL, json=payload, headers=headers)
response_json = response.json()
# Extract text response
return response_json.get("model_response", {}).get("content", [{}])[0].get("text", "🤖 No response received.")
except Exception as e:
return f"⚠️ Error: {str(e)}"
|