Initial commit

app.py

@@ -50,6 +50,7 @@ def trait_classifier(text: str) -> Dict[str, float]:
 
     # Format output for Gradio (e.g. {"label_1": probability_1, "label_2": probability_2})
     output_dict = {}
+    output_dict['Personality traits'] = 1
     #agr
     if outputs_agr[0]['label'] == 'POSITIVE':
         output_dict['Agreeablenes'] = outputs_agr[0]["score"]
@@ -96,13 +97,14 @@ Fine-tuned from [xlm-roberta-base](https://huggingface.co/FacebookAI/xlm-roberta
 
 demo = gr.Interface(fn=trait_classifier, 
              inputs="text", 
-             outputs=gr.Label(num_top_classes=5, label='Traits'),
+             outputs=gr.Label(num_top_classes=5, label='output'),
              title="🤯 Petarda",
-             theme="gr.themes.Ocean()",
+             theme=gr.themes.Ocean(primary_hue="amber", secondary_hue="cyan"),
              description=description,
              examples=[["I have a real problem right now... I am struggling a lot :("],
-                       ["He's such a nice, gentle man and it was great"]])
+                       ["He's such a nice, gentle man and it was great"],
+                       "And now you\'re asking very hard questions that I\'ve struggled with quite a bit… Both of those can be a struggle. My memory is very hard to access because I just can\'t remember things in any level of detail... faces, conversations, events, my own actions, or unstructured facts. My senses can be hard to access as well because I keep pulling back into my own head. I don\t feel like I\'m grounded in myself or that I have a consistent behavior. I\'m stateless. It\'s as if everything is new to me all the time. I can barely maintain a routine. At the same time, empathizing and really knowing what to do naturally in social situations is really a problem for me as well. I seem to get by and people generally like me, but I\'m very careful and I can\'t always relate to how other people are acting. Sometimes it seems animalistic, simple-minded, and petty. Depends... if there\'s something that I\'m interested in being discussed or done, then I am very, very energetic, passionate, and reactive. My energy levels can swing from one extreme to another depending on my level of interest. If not, then I\'m extremely withdrawn and in my own head."])
 
 # 4. Launch the interface
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)