tomsoderlund commited on
Commit
faec829
·
1 Parent(s): 98314f0

Gradio examples

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +3 -3
README.md CHANGED
@@ -37,4 +37,4 @@ and test your Gradio app on: http://127.0.0.1:7860/
37
 
38
  ## REST API via Gradio “Use via API” (see page footer)
39
 
40
- curl -X POST -H 'Content-type: application/json' --data '{ "data": ["Wirwachs malmgård är en 1770-talet uppförd egendom som ligger på västra Södermalm i Stockholm."] }' https://tomsoderlund-swedish-entity-recognition.hf.space/run/predict
 
37
 
38
  ## REST API via Gradio “Use via API” (see page footer)
39
 
40
+ curl -X POST -H 'Content-type: application/json' --data '{ "data": ["Jag heter Tom och bor i Stockholm."] }' https://tomsoderlund-swedish-entity-recognition.hf.space/run/predict
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import gradio as gr
2
  from transformers import pipeline
3
 
4
  def process_swedish_text(text):
@@ -12,7 +12,7 @@ def process_swedish_text(text):
12
  nlp_results_adjusted = map(lambda entity: entity | { 'score': float(entity['score']) }, nlp_results)
13
  print('nlp_results_adjusted:', nlp_results_adjusted)
14
  # Return values
15
- return str({'entities': list(nlp_results_adjusted)})
16
 
17
- gradio_interface = gr.Interface(fn=process_swedish_text, inputs="text", outputs="json")
18
  gradio_interface.launch()
 
1
+ import gradio
2
  from transformers import pipeline
3
 
4
  def process_swedish_text(text):
 
12
  nlp_results_adjusted = map(lambda entity: entity | { 'score': float(entity['score']) }, nlp_results)
13
  print('nlp_results_adjusted:', nlp_results_adjusted)
14
  # Return values
15
+ return {'entities': list(nlp_results_adjusted)}
16
 
17
+ gradio_interface = gradio.Interface(fn=process_swedish_text, inputs="text", outputs="json", examples=[["Jag heter Tom och bor i Stockholm."], ["Groens malmgård är en av Stockholms malmgårdar, belägen vid Malmgårdsvägen 53 på Södermalm i Stockholm."]])
18
  gradio_interface.launch()