| |
| """ |
| SamyamLM-V1 — Test Inference Client |
| Quickly run spatial & satellite image analysis with local Ollama runtime. |
| """ |
|
|
| import sys |
| import base64 |
| import requests |
|
|
| def analyze_image(image_path: str, prompt: str = "Analyze this image and identify all objects with bounding boxes."): |
| |
| with open(image_path, "rb") as f: |
| img_b64 = base64.b64encode(f.read()).decode("utf-8") |
|
|
| |
| payload = { |
| "model": "samyam-lm-v1", |
| "prompt": prompt, |
| "images": [img_b64], |
| "stream": False, |
| "options": { |
| "temperature": 0.2, |
| "top_p": 0.9 |
| } |
| } |
|
|
| try: |
| response = requests.post("http://localhost:11434/api/generate", json=payload) |
| response.raise_for_status() |
| return response.json().get("response", "") |
| except requests.exceptions.ConnectionError: |
| return "❌ Error: Ollama is not running on http://localhost:11434. Start it with: ollama serve" |
|
|
| if __name__ == "__main__": |
| if len(sys.argv) < 2: |
| print("Usage: python inference.py <path_to_satellite_or_drone_image.jpg> [prompt]") |
| sys.exit(1) |
|
|
| img_path = sys.argv[1] |
| user_prompt = sys.argv[2] if len(sys.argv) > 2 else "Describe tactical assets and provide bounding boxes [ymin, xmin, ymax, xmax]." |
| |
| print(f"🛰️ Running SamyamLM-V1 inference on: {img_path}...") |
| result = analyze_image(img_path, user_prompt) |
| print("\n--- Model Output ---") |
| print(result) |
|
|
