Spaces:
Build error
Build error
Update main.py
Browse files
main.py
CHANGED
|
@@ -8,7 +8,6 @@ llm = AutoModelForCausalLM.from_pretrained(
|
|
| 8 |
max_new_tokens=1096,
|
| 9 |
threads=3,
|
| 10 |
)
|
| 11 |
-
|
| 12 |
# Pydantic object
|
| 13 |
class Validation(BaseModel):
|
| 14 |
user_prompt: str # User's prompt
|
|
|
|
| 8 |
max_new_tokens=1096,
|
| 9 |
threads=3,
|
| 10 |
)
|
|
|
|
| 11 |
# Pydantic object
|
| 12 |
class Validation(BaseModel):
|
| 13 |
user_prompt: str # User's prompt
|