Merge branch 'main' of https://huggingface.co/spaces/burtenshaw/disticleaner

app.py

@@ -1,3 +1,5 @@
+import spaces
+
 import os
 
 import gradio as gr
@@ -5,24 +7,20 @@ import gradio as gr
 
 log_file = "out-file.txt"
 
-
+@spaces.GPU(duration=120)
 def run():
     import subprocess
     import sys
 
-    with open("out-file.txt", "w") as f:
-        process = subprocess.Popen(
-            [f"{sys.executable}", "prometheus_pipeline.py"],
-            stdout=subprocess.PIPE,
-        )
+    process = subprocess.Popen(
+        [f"{sys.executable}", "prometheus_pipeline.py"],
+        stdout=subprocess.PIPE,
+    )
 
     content = ""
 
     for line in iter(process.stdout.readline, ""):
-        content = content.split("\n")
-        content.append(line.decode("utf-8"))
-        content = content[-10:]
-        content = "\n".join(content)
+        content += line.decode("utf-8") + "\n"
         yield content