updated cors

.ipynb_checkpoints/app-checkpoint.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, File, UploadFile
+from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 from transformers import AutoProcessor, Blip2ForConditionalGeneration
 import torch
@@ -6,6 +7,14 @@ import io
 
 app = FastAPI()
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Adjust this as needed for security
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # Load the model and processor
 model = Blip2ForConditionalGeneration.from_pretrained("ybelkada/blip2-opt-2.7b-fp16-sharded")
 model.load_adapter('blip-cpu-model')

.ipynb_checkpoints/streamlit_app-checkpoint.py

@@ -11,7 +11,7 @@ if uploaded_file is not None:
     st.image(image, caption="Uploaded Image", use_column_width=True)
     
     files = {"file": uploaded_file.getvalue()}
-    response = requests.post("http://localhost:8502/generate-caption/", files=files)
+    response = requests.post("http://0.0.0.0:8502/generate-caption/", files=files)
     caption = response.json().get("caption")
     
     st.write("Generated Caption:")

app.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, File, UploadFile
+from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 from transformers import AutoProcessor, Blip2ForConditionalGeneration
 import torch
@@ -6,6 +7,14 @@ import io
 
 app = FastAPI()
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Adjust this as needed for security
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # Load the model and processor
 model = Blip2ForConditionalGeneration.from_pretrained("ybelkada/blip2-opt-2.7b-fp16-sharded")
 model.load_adapter('blip-cpu-model')

streamlit_app.py

@@ -11,7 +11,8 @@ if uploaded_file is not None:
     st.image(image, caption="Uploaded Image", use_column_width=True)
     
     files = {"file": uploaded_file.getvalue()}
-    response = requests.post("http://localhost:8502/generate-caption/", files=files)
+    print("Sending API request")
+    response = requests.post("http://0.0.0.0:8502/generate-caption/", files=files)
     caption = response.json().get("caption")
     
     st.write("Generated Caption:")