OFA-Sys/OFA-Image_Caption

README.md

@@ -21,5 +21,6 @@ gradio
 `app_file`: _string_
 app.py
 
+
 `pinned`: _boolean_
 false

app.py

@@ -1,7 +1,7 @@
 import os
 
 os.system('cd fairseq;'
-          'pip install ./; cd ..')
+          'pip install --use-feature=in-tree-build ./; cd ..')
 os.system('ls -l')
 
 import torch
@@ -120,4 +120,4 @@ examples = [['beatles.jpeg'], ['aurora.jpeg'], ['good_luck.png'], ['pokemons.jpg
 io = gr.Interface(fn=image_caption, inputs=gr.inputs.Image(type='pil'), outputs=gr.outputs.Textbox(label="Caption"),
                   title=title, description=description, article=article, examples=examples,
                   allow_flagging=False, allow_screenshot=False)
-io.launch()
+io.launch(cache_examples=True)

fairseq/fairseq/data/indexed_dataset.py

@@ -118,7 +118,7 @@ _code_to_dtype = {
     3: np.int16,
     4: np.int32,
     5: np.int64,
-    6: np.float64,
+    6: np.float,
     7: np.double,
     8: np.uint16,
     9: np.uint32,
@@ -325,7 +325,7 @@ class IndexedDatasetBuilder:
         np.int16: 2,
         np.int32: 4,
         np.int64: 8,
-        np.float64: 4,
+        np.float: 4,
         np.double: 8,
     }