Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ class GenerateRequest(BaseModel):
|
|
| 63 |
المادة: str
|
| 64 |
المستوى: str
|
| 65 |
|
| 66 |
-
@app.post("/generate
|
| 67 |
def generate_text(request: GenerateRequest):
|
| 68 |
المادة = request.المادة
|
| 69 |
المستوى = request.المستوى
|
|
@@ -80,7 +80,7 @@ def generate_text(request: GenerateRequest):
|
|
| 80 |
logger.info(f"Formatted prompt: {arabic_prompt}")
|
| 81 |
|
| 82 |
# Tokenize the prompt
|
| 83 |
-
inputs = tokenizer(arabic_prompt, return_tensors="pt", max_length=
|
| 84 |
|
| 85 |
# Generate text
|
| 86 |
with torch.no_grad():
|
|
|
|
| 63 |
المادة: str
|
| 64 |
المستوى: str
|
| 65 |
|
| 66 |
+
@app.post("/generate")
|
| 67 |
def generate_text(request: GenerateRequest):
|
| 68 |
المادة = request.المادة
|
| 69 |
المستوى = request.المستوى
|
|
|
|
| 80 |
logger.info(f"Formatted prompt: {arabic_prompt}")
|
| 81 |
|
| 82 |
# Tokenize the prompt
|
| 83 |
+
inputs = tokenizer(arabic_prompt, return_tensors="pt", max_length=1024, truncation=True)
|
| 84 |
|
| 85 |
# Generate text
|
| 86 |
with torch.no_grad():
|