BugFix: Fix permissions for downloaded files

Author: CodeWithKyrian

src/Commands/DownloadModelCommand.php

@@ -9,13 +9,15 @@
 use Codewithkyrian\Transformers\Models\Auto\AutoModelForCausalLM;
 use Codewithkyrian\Transformers\Models\Auto\AutoModelForSeq2SeqLM;
 use Codewithkyrian\Transformers\Models\Auto\AutoModelForSequenceClassification;
+use Codewithkyrian\Transformers\Pipelines\Task;
 use Codewithkyrian\Transformers\Transformers;
 use Symfony\Component\Console\Attribute\AsCommand;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
+use function Codewithkyrian\Transformers\Pipelines\pipeline;
 
 #[AsCommand(
     name: 'download-model',
@@ -63,13 +65,14 @@ protected function execute(InputInterface $input, OutputInterface $output): int
 
         // Download the model
         try {
-            // TODO: Verify the tasks and corresponding AutoModel classes
-            $model = match ($task) {
-                'text-generation' => AutoModelForCausalLM::fromPretrained($model, $quantized, cacheDir: $cacheDir),
-                'text-classification', 'sentiment-analysis' => AutoModelForSequenceClassification::fromPretrained($model, $quantized, cacheDir: $cacheDir),
-                'translation' => AutoModelForSeq2SeqLM::fromPretrained($model, $quantized, cacheDir: $cacheDir),
-                default => AutoModel::fromPretrained($model, $quantized, cacheDir: $cacheDir),
-            };
+            $task = Task::tryFrom($task);
+
+            if ($task != null) {
+                pipeline($task, $model);
+            } else {
+                AutoModel::fromPretrained($model, $quantized, cacheDir: $cacheDir);
+            }
+
 
             $output->writeln('✔ Model downloaded successfully.');
 

src/Utils/Hub.php

@@ -120,6 +120,8 @@ public static function getFile(
                 rename($partPath, $filePath);
             }
 
+            chmod($filePath, 0644);
+
             return $filePath;
         } catch (GuzzleException $e) {
             self::handleException($e->getCode(), $remoteURL, $fatal);