Added requirements and removed Device for CPU only

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-import torch
 from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM, AutoConfig
 
@@ -9,17 +8,16 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
 # Function to load the model and processor
 @st.cache_resource
 def load_model_and_processor():
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     config = AutoConfig.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
     config.vision_config.model_type = "davit"
-    model = AutoModelForCausalLM.from_pretrained("sujet-ai/Lutece-Vision-Base", config=config, trust_remote_code=True).to(device).eval()
+    model = AutoModelForCausalLM.from_pretrained("sujet-ai/Lutece-Vision-Base", config=config, trust_remote_code=True).eval()
     processor = AutoProcessor.from_pretrained("sujet-ai/Lutece-Vision-Base", config=config, trust_remote_code=True)
-    return model, processor, device
+    return model, processor
 
 # Function to generate answer
-def generate_answer(model, processor, device, image, prompt):
+def generate_answer(model, processor, image, prompt):
     task = "<FinanceQA>"
-    inputs = processor(text=prompt, images=image, return_tensors="pt").to(device)
+    inputs = processor(text=prompt, images=image, return_tensors="pt")
     generated_ids = model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],
@@ -54,7 +52,7 @@ def main():
     st.sidebar.markdown("Our website : [sujet.ai](https://sujet.ai)")
 
     # Load model and processor
-    model, processor, device = load_model_and_processor()
+    model, processor = load_model_and_processor()
 
     # File uploader for document
     uploaded_file = st.file_uploader("📄 Upload a financial document", type=["png", "jpg", "jpeg"])
@@ -68,7 +66,7 @@ def main():
 
         if st.button("🔍 Generate Answer"):
             with st.spinner("Generating answer..."):
-                answer = generate_answer(model, processor, device, image, question)
+                answer = generate_answer(model, processor, image, question)
             st.success(f"## 💡 {answer}")
 
     # # Model configuration viewer

requirements.txt

@@ -0,0 +1,5 @@
+timm
+transformers
+spaces
+pillow
+torch