adds pt dep

app.py

@@ -13,4 +13,4 @@ def generate_solution(prompt, **kwargs):
     output = model.generate(input_ids, pad_token_id=tokenizer.pad_token_id, *kwargs)
     return tokenizer.decode(output[0][start:])
 
-gr.Interface(generate_solution, inputs="text", outputs="text").launch()
+gr.Interface(generate_solution, inputs="text", outputs="text").launch()

requirements.txt

@@ -1 +1,2 @@
+torch
 transformers