Csplk commited on
Commit
dea0b92
·
1 Parent(s): c81016f

fix missing close string

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -361,7 +361,7 @@ class InferenceNodeGenerator(NodeGenerator):
361
  "text2text-generation": {"generated_text": ("str", "gr.Textbox(label='Output')")},
362
  "summarization": {"summary": ("str", "gr.Textbox(label='Summary')")},
363
  "translation": {"translation": ("str", "gr.Textbox(label='Translation')")},
364
- "question-answering": {"answer": ("str", "gr.Textbox(label='Answer'))},
365
  "image-classification": {"labels": ("list", "gr.JSON(label='Predictions')")},
366
  "object-detection": {"objects": ("list", "gr.JSON(label='Detections')")},
367
  "image-segmentation": {"masks": ("list", "gr.JSON(label='Segments')")},
@@ -369,7 +369,7 @@ class InferenceNodeGenerator(NodeGenerator):
369
  "image-to-text": {"text": ("str", "gr.Textbox(label='Description')")},
370
  "automatic-speech-recognition": {"text": ("str", "gr.Textbox(label='Transcription')")},
371
  "text-to-speech": {"audio": ("filepath", "gr.Audio(label='Generated Audio')")},
372
- "zero-shot-classification": {"scores": ("list", "gr.JSON(label='Scores'))},
373
  }
374
 
375
  def get_model_info(self, model_id: str) -> Optional[Dict]:
 
361
  "text2text-generation": {"generated_text": ("str", "gr.Textbox(label='Output')")},
362
  "summarization": {"summary": ("str", "gr.Textbox(label='Summary')")},
363
  "translation": {"translation": ("str", "gr.Textbox(label='Translation')")},
364
+ "question-answering": {"answer": ("str", "gr.Textbox(label='Answer')")},
365
  "image-classification": {"labels": ("list", "gr.JSON(label='Predictions')")},
366
  "object-detection": {"objects": ("list", "gr.JSON(label='Detections')")},
367
  "image-segmentation": {"masks": ("list", "gr.JSON(label='Segments')")},
 
369
  "image-to-text": {"text": ("str", "gr.Textbox(label='Description')")},
370
  "automatic-speech-recognition": {"text": ("str", "gr.Textbox(label='Transcription')")},
371
  "text-to-speech": {"audio": ("filepath", "gr.Audio(label='Generated Audio')")},
372
+ "zero-shot-classification": {"scores": ("list", "gr.JSON(label='Scores')")},
373
  }
374
 
375
  def get_model_info(self, model_id: str) -> Optional[Dict]: