Upload 3 files

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,58 @@
+from fastapi import FastAPI, UploadFile, File
+from fastapi.middleware.cors import CORSMiddleware
+import google.generativeai as genai
+from PIL import Image
+import io
+import os
+from dotenv import load_dotenv
+
+
+load_dotenv()
+
+
+app = FastAPI(title="Simple Image Description API")
+
+
+origins = ["*"]
+
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+genai.configure(api_key=os.getenv("GOOGLE_GEMINI_API_KEY"))
+
+
+model = genai.GenerativeModel('gemini-1.5-flash')
+
+
+@app.post("/describe-image/")
+async def describe_image(imageInputByUser: UploadFile = File(...)):
+
+    try:
+
+        image_content = await imageInputByUser.read()
+        
+
+        img = Image.open(io.BytesIO(image_content))
+        
+
+        response = model.generate_content(["Please provide a short description of this image", img])
+        
+
+        return {
+            'success': True,
+            'message': response.text
+        }
+    
+    except Exception as e:
+
+        return {
+            'success': False,
+            'message': str(e)
+        }

requirements.txt

@@ -0,0 +1,6 @@
+fastapi==0.109.2
+Pillow==11.0.0
+python-dotenv==1.0.1
+python-multipart==0.0.9
+google-generativeai==0.7.1
+uvicorn==0.27.1