yakine commited on
Commit
9c4411c
·
verified ·
1 Parent(s): 1a24301

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=512, truncation=True)
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():