Spaces:
Running
Running
File size: 1,033 Bytes
e078b1d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | from __future__ import annotations
from pydantic import BaseModel, Field
from typing import List
class SummarizeRequest(BaseModel):
text: str = Field(..., min_length=10)
model_choice: str = Field(default="bart_large_cnn")
max_length: int = Field(default=96, ge=16, le=512)
dataset_track: str = Field(default="gcc")
class SummarizeResponse(BaseModel):
model_name: str
summary: str
dataset_track: str
word_count: int
class CompareRequest(BaseModel):
text: str = Field(..., min_length=10)
model_choices: List[str]
max_length: int = Field(default=96, ge=16, le=512)
dataset_track: str = Field(default="gcc")
class CompareResponseItem(BaseModel):
model_name: str
summary: str
word_count: int
class CompareResponse(BaseModel):
dataset_track: str
items: List[CompareResponseItem]
class SampleItem(BaseModel):
id: str
dataset_track: str
title: str
text: str
source_label: str
class SamplesResponse(BaseModel):
items: List[SampleItem]
|