Update app.py

app.py

@@ -5,18 +5,18 @@ import logging
 import time
 import os
 from datetime import datetime
-from typing import List, Dict, Any
+from typing import List, Dict
 import requests
 import gradio as gr
 import atexit
 import subprocess
-from urllib.parse import urlparse, quote
-import warnings
 import webbrowser
-from flask import Flask, request, jsonify
-from threading import Thread
-
-app = Flask(__name__)
+import urllib.parse
+import warnings
+import torch
+import flask
+import spaces
+from accelerate import Accelerator
 
 # Constants
 INPUT_DIRECTORY = 'input'
@@ -25,12 +25,6 @@ LOGS_DIRECTORY = 'logs'
 RESOLUTIONS_DIRECTORY = 'resolutions'
 REPOS_DIRECTORY = 'repos'
 
-# Set up environment
-def initialize_environment(input_file, output_directory):
-    directories = [LOGS_DIRECTORY, RESOLUTIONS_DIRECTORY, REPOS_DIRECTORY, input_file, output_directory]
-    for directory in directories:
-        os.makedirs(directory, exist_ok=True)
-
 # Set up logging
 def initialize_logger():
     log_file = f"{LOGS_DIRECTORY}/github_bot_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
@@ -44,9 +38,13 @@ def initialize_logger():
     )
     return logging.getLogger(__name__)
 
-# Initialize environment and logger
-initialize_environment(INPUT_DIRECTORY, OUTPUT_DIRECTORY)
-logger = initialize_logger()  # Initialize logger globally
+# Set up environment
+@spaces.GPU
+def initialize_environment(input_file, output_directory):
+    directories = [LOGS_DIRECTORY, RESOLUTIONS_DIRECTORY, REPOS_DIRECTORY, input_file, output_directory]
+    for directory in directories:
+        os.makedirs(directory, exist_ok=True)
+    return initialize_logger()
 
 # GitHub API handler
 class GitHubAPI: