Spaces:
Running
Running
fix missing close string
Browse files
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]:
|