add OLM model, rem depracted gradio 'type' arg

.gitignore

@@ -1,4 +1,4 @@
 venv_*
 __pycache__*
 .DS_Store
-
+app.py-*

README.md

@@ -9,4 +9,11 @@ app_file: app.py
 pinned: false
 ---
 
+# Setup env:
+```
+python3 -m venv venv_llm
+source venv_llm/bin/activate
+pip install -r requirements.txt
+```
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -17,6 +17,7 @@ MODEL_NAME_DICT = {
     "bert-large-uncased": "BERT-large",
     "roberta-base": "RoBERTa-base",
     "bert-base-uncased": "BERT-base",
+    "olm/olm-roberta-base-oct-2022": "OLM_RoBERTa-base",
     OWN_MODEL_NAME: "Your model's"
 }
 MODEL_NAMES = list(MODEL_NAME_DICT.keys())
@@ -50,6 +51,7 @@ GENDERED_LIST = [
 ]
 
 
+
 # %%
 # Fire up the models
 models = {m : pipeline("fill-mask", model=m) for m in MODEL_NAMES if m != OWN_MODEL_NAME}
@@ -265,12 +267,11 @@ with demo:
         "If there is an * by a sentence number, then at least one top prediction for that sentence was non-gendered.")
 
     with gr.Row():
-        female_fig = gr.Plot(type="auto")
+        female_fig = gr.Plot()#type="auto")
     with gr.Row():
         female_df = gr.Dataframe()
     with gr.Row():
-        display_text = gr.Textbox(
-            type="auto", label="Sample of text fed to model")
+        display_text = gr.Textbox(label="Sample of text fed to model")
 
     uncertain_btn.click(
         fn=predict_gender_pronouns,
@@ -279,6 +280,6 @@ with demo:
         outputs=[display_text, female_df, female_fig]
     )
 
-demo.launch(debug=True)
+demo.launch(share=True, debug=True)
 
 # %%