ipvikas commited on
Commit
4a129f4
·
1 Parent(s): 3cb15b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -27,18 +27,17 @@ feature_extractor = ViTFeatureExtractor.from_pretrained(repo_name)
27
  tokenizer = AutoTokenizer.from_pretrained(repo_name)
28
  model = VisionEncoderDecoderModel.from_pretrained(repo_name)
29
 
30
- def get_quote(image_1):
31
  #reader = easyocr.Reader(['en'])
32
  #image = Image.open(image_1,mode = 'r')
33
  #image = Image.open(image_1)
34
 
35
- image_2 = Image.open(requests.get(image_1, stream=True).raw)
36
- IMAGE_SHAPE = (224, 224)#(321, 321)
 
 
 
37
 
38
- #IMAGE_SHAPE+(3,)
39
- image = image_2.resize(IMAGE_SHAPE)
40
- #image.save("cats.png")
41
- #image
42
 
43
 
44
 
 
27
  tokenizer = AutoTokenizer.from_pretrained(repo_name)
28
  model = VisionEncoderDecoderModel.from_pretrained(repo_name)
29
 
30
+ def get_quote(image_path):
31
  #reader = easyocr.Reader(['en'])
32
  #image = Image.open(image_1,mode = 'r')
33
  #image = Image.open(image_1)
34
 
35
+ #image_2 = Image.open(requests.get(image_1, stream=True).raw)
36
+
37
+ image = Image.open(image_path)
38
+
39
+
40
 
 
 
 
 
41
 
42
 
43