MohamedRashad commited on
Commit
f157a13
Β·
1 Parent(s): 9865789

Refactor app.py and requirements.txt: Reorganize imports and add 'accelerate' to dependencies

Browse files
Files changed (2) hide show
  1. app.py +5 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,18 +1,22 @@
 
1
  import json
2
  import math
3
  import os
4
  import traceback
5
  from io import BytesIO
6
  from typing import Any, Dict, List, Optional, Tuple
7
- from huggingface_hub import snapshot_download
8
  import fitz # PyMuPDF
9
  import gradio as gr
10
  import requests
11
  import torch
 
12
  from PIL import Image, ImageDraw, ImageFont
13
  from qwen_vl_utils import process_vision_info
14
  from transformers import AutoModelForCausalLM, AutoProcessor
15
 
 
 
16
  # Constants
17
  MIN_PIXELS = 3136
18
  MAX_PIXELS = 11289600
 
1
+ import spaces
2
  import json
3
  import math
4
  import os
5
  import traceback
6
  from io import BytesIO
7
  from typing import Any, Dict, List, Optional, Tuple
8
+
9
  import fitz # PyMuPDF
10
  import gradio as gr
11
  import requests
12
  import torch
13
+ from huggingface_hub import snapshot_download
14
  from PIL import Image, ImageDraw, ImageFont
15
  from qwen_vl_utils import process_vision_info
16
  from transformers import AutoModelForCausalLM, AutoProcessor
17
 
18
+ print(torch.__version__)
19
+
20
  # Constants
21
  MIN_PIXELS = 3136
22
  MAX_PIXELS = 11289600
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  huggingface_hub
 
2
  transformers
3
  torch
4
  torchvision
 
1
  huggingface_hub
2
+ accelerate
3
  transformers
4
  torch
5
  torchvision