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():
|