Update app.py

app.py

@@ -11,15 +11,15 @@ import torch
 import tempfile
 import whisper
 import json
-import uvicorn
+#import uvicorn
 app = Flask(__name__)
 from deep_translator import GoogleTranslator
 
 CORS(app, resources={r"/*": {"origins": "*"}})  # Enable CORS for all routes
 # Load MedGemma model (4B) on startup
-from asgiref.wsgi import WsgiToAsgi
+#from asgiref.wsgi import WsgiToAsgi
 
-asgi_app = WsgiToAsgi(app)  # for uvicorn compatibility
+#asgi_app = WsgiToAsgi(app)  # for uvicorn compatibility
 from transformers import AutoTokenizer, AutoModelForCausalLM
 #from huggingface_hub import login
 import os
@@ -46,7 +46,7 @@ model_id = "google/medgemma-4b-pt"
 model_medg = AutoModelForImageTextToText.from_pretrained(
     model_id,
     torch_dtype=torch.bfloat16,
-    device_map="auto",
+    device_map="cpu",
 )
 processor = AutoProcessor.from_pretrained(model_id)
 @app.route('/analyze-image', methods=['POST'])